diff --git a/src/Models/Brand.php b/src/Models/Brand.php
index d94001d..9d77e8c 100644
--- a/src/Models/Brand.php
+++ b/src/Models/Brand.php
@@ -36,7 +36,7 @@ protected static function booting()
                 })
                 ->leftJoinSub($renamedStore, 'store_value', function ($join) {
                     $join->on('store_value.sub_option_id', '=', 'eav_attribute_option.option_id')
-                         ->where('store_value.store_id', Site::selected()->attributes['magento_store_id']);
+                         ->where('store_value.store_id', Site::current()->attributes['magento_store_id']);
                 })
                 ->where('attribute_id', config('rapidez.statamic.runway.brand_attribute_id'));
         });
diff --git a/src/Models/Category.php b/src/Models/Category.php
index 49b39ac..69d72e8 100644
--- a/src/Models/Category.php
+++ b/src/Models/Category.php
@@ -14,6 +14,6 @@ class Category extends Model
 
     public function getTable()
     {
-        return 'catalog_category_flat_store_'.Site::selected()->attributes['magento_store_id'];
+        return 'catalog_category_flat_store_'.Site::current()->attributes['magento_store_id'];
     }
 }
diff --git a/src/Models/Product.php b/src/Models/Product.php
index 7c0fb88..d1e396b 100644
--- a/src/Models/Product.php
+++ b/src/Models/Product.php
@@ -23,6 +23,6 @@ protected static function booting()
 
     public function getTable()
     {
-        return 'catalog_product_flat_'.Site::selected()->attributes['magento_store_id'];
+        return 'catalog_product_flat_'.Site::current()->attributes['magento_store_id'];
     }
 }