diff --git a/scripts/generate_db_helpers.go b/scripts/generate_db_helpers.go
index 3f4e68483f090ad141b947f65d73bcdb12e0e255_c2NyaXB0cy9nZW5lcmF0ZV9kYl9oZWxwZXJzLmdv..714427040ae353f3a42327aa8043ccff73d8fdf5_c2NyaXB0cy9nZW5lcmF0ZV9kYl9oZWxwZXJzLmdv 100644
--- a/scripts/generate_db_helpers.go
+++ b/scripts/generate_db_helpers.go
@@ -81,6 +81,13 @@
 	if err != nil {
 		log.Fatal(err)
 	}
+	if len(packages) == 2 {
+		for k := range packages {
+			if strings.HasSuffix(k, "_test") {
+				delete(packages, k)
+			}
+		}
+	}
 	if len(packages) != 1 {
 		log.Fatal("Expected to find 1 go package, got: ", len(packages), packages)
 	}