diff --git a/test/testit.go b/test/testit.go
index 2229f89..9d90098 100644
--- a/test/testit.go
+++ b/test/testit.go
@@ -190,7 +190,7 @@ func importProduct(product models.Product, db *gorm.DB) {
 		Sku:               product.ProductGroupID,
 		Type:              "simple",
 		AttributeFamilyID: families[0].ID,
-		Brand:             brand,
+		BrandID:           brand.ID,
 		Categories:        categories,
 		AttributeValues:   attributes,
 	}
@@ -211,16 +211,21 @@ func importProduct(product models.Product, db *gorm.DB) {
 		iproduct.Type = "configurable"
 		iproduct.AttributeFamilyID = families[1].ID
 	}
-	//todo insert attribute values manytomany
+
+	errMainProduct := db.Omit("Categories.*", "SuperAttributes.*", "ParentID").Create(&iproduct).Error
+	if errMainProduct != nil {
+		log.Println(errMainProduct)
+		return
+	}
+
 	mainProductFlat.Product = iproduct
-	mainProductFlat.ProductID = iproduct.ID
 	mainProductFlat.BrandID = brand.ID
 	mainProductFlat.MetaKeywords = keywords
 
-	errProduct := db.Omit("Product.*").Create(&mainProductFlat).Error
+	errProductMainFlat := db.Omit("Product.*").Create(&mainProductFlat).Error
 
-	if errProduct != nil {
-		log.Println(errProduct)
+	if errProductMainFlat != nil {
+		log.Println(errProductMainFlat)
 		return
 	}