diff --git a/packages/Webkul/Velocity/src/Http/Controllers/Shop/ComparisonController.php b/packages/Webkul/Velocity/src/Http/Controllers/Shop/ComparisonController.php
index 2326da6af..f95fcb5cd 100644
--- a/packages/Webkul/Velocity/src/Http/Controllers/Shop/ComparisonController.php
+++ b/packages/Webkul/Velocity/src/Http/Controllers/Shop/ComparisonController.php
@@ -76,10 +76,22 @@ class ComparisonController extends Controller
 
         if (! $compareProduct) {
             // insert new row
-            $this->compareProductsRepository->create([
-                'customer_id'     => $customerId,
-                'product_flat_id' => $productId,
-            ]);
+
+            $productFlatRepository = app('\Webkul\Product\Models\ProductFlat');
+
+            $productFlat = $productFlatRepository
+                            ->where('product_id', $productId)
+                            ->get()
+                            ->first();
+
+            if ($productFlat) {
+                $productId = $productFlat->id;
+                
+                $this->compareProductsRepository->create([
+                    'customer_id'     => $customerId,
+                    'product_flat_id' => $productId,
+                ]);
+            }
 
             return response()->json([
                 'status'  => 'success',