diff --git a/yb-voyager/cmd/importDataFileTaskPicker_test.go b/yb-voyager/cmd/importDataFileTaskPicker_test.go
index c24a17ffc..44b9f1ed8 100644
--- a/yb-voyager/cmd/importDataFileTaskPicker_test.go
+++ b/yb-voyager/cmd/importDataFileTaskPicker_test.go
@@ -1,3 +1,4 @@
+//go:build unit
 /*
 Copyright (c) YugabyteDB, Inc.
 
diff --git a/yb-voyager/src/query/sqltransformer/transformer.go b/yb-voyager/src/query/sqltransformer/transformer.go
index b716e97af..4c2ec46fe 100644
--- a/yb-voyager/src/query/sqltransformer/transformer.go
+++ b/yb-voyager/src/query/sqltransformer/transformer.go
@@ -1,3 +1,4 @@
+//go:build unit
 /*
 Copyright (c) YugabyteDB, Inc.