diff --git a/pkg/db/connect_test.go b/pkg/db/connect_test.go
index 11c5df2..6b58727 100644
--- a/pkg/db/connect_test.go
+++ b/pkg/db/connect_test.go
@@ -23,6 +23,7 @@ func TestCreateAndConnectToDB(t *testing.T) {
 	db.Init(&schema_sql, &migrations)
 
 	data_dir := "../../sample_data/data"
+	_ = os.MkdirAll(data_dir, os.FileMode(0o644))
 	// Create then connect to a new empty DB
 	dbPath := filepath.Join(data_dir, "test.db")
 	_ = os.Remove(dbPath) // Delete it if it exists
diff --git a/pkg/sqlgenerate/test_schemas/food.sql b/sample_data/test_schemas/food.sql
similarity index 100%
rename from pkg/sqlgenerate/test_schemas/food.sql
rename to sample_data/test_schemas/food.sql