diff --git a/packages/Webkul/Product/src/Helpers/ProductImage.php b/packages/Webkul/Product/src/Helpers/ProductImage.php
index 778995166..16c906027 100755
--- a/packages/Webkul/Product/src/Helpers/ProductImage.php
+++ b/packages/Webkul/Product/src/Helpers/ProductImage.php
@@ -52,12 +52,9 @@ class ProductImage extends AbstractProduct
      */
     public function getProductBaseImage($product)
     {
-        if (! $product) 
-            return [];
-        
-        $images = $product->images;
+        $images = $product ? $product->images : null;
 
-        if ($images->count()) {
+        if ($images && $images->count()) {
             $image = [
                 'small_image_url' => url('cache/small/' . $images[0]->path),
                 'medium_image_url' => url('cache/medium/' . $images[0]->path),