Fix: register m2m models

baking-bad · Sep 28, 2023 · 062a5c2 · 062a5c2
1 parent 58323fc
commit 062a5c2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/postgres/core/postgres.go b/internal/postgres/core/postgres.go
@@ -96,6 +96,9 @@ func New(cfg Config, schemaName, appName string, opts ...PostgresOption) (*Postg
 		opt(&postgres)
 	}
 
+	// register many-to-many relationships
+	postgres.DB.RegisterModel(models.ManyToMany()...)
+
 	return &postgres, nil
 }
 
@@ -125,9 +128,6 @@ func WaitNew(cfg Config, schemaName, appName string, timeout int, opts ...Postgr
 		time.Sleep(time.Second * time.Duration(timeout))
 	}
 
-	// register many-to-many relationships
-	db.DB.RegisterModel(models.ManyToMany()...)
-
 	return db
 }