diff --git a/test/testit.go b/test/testit.go
index 6672af1..b723677 100644
--- a/test/testit.go
+++ b/test/testit.go
@@ -213,10 +213,11 @@ func importProduct(product models.Product, db *gorm.DB) {
 	}
 	//todo insert attribute values manytomany
 	mainProductFlat.Product = iproduct
-	mainProductFlat.Brand = brand
+	mainProductFlat.ProductID = iproduct.ID
+	mainProductFlat.BrandID = brand.ID
 	mainProductFlat.MetaKeywords = keywords
 
-	errProduct := db.Omit("Product.ParentID", "Product.SuperAttributes.*", "Product.Categories.*").Create(&mainProductFlat).Error
+	errProduct := db.Omit("Product.*").Create(&mainProductFlat).Error
 
 	if errProduct != nil {
 		log.Println(errProduct)