small fixes

hivesolutions · Aug 27, 2018 · fdc0e4d · fdc0e4d
1 parent 66a8194
commit fdc0e4d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/budy/models/product.py b/src/budy/models/product.py
@@ -616,8 +616,8 @@ def related(self, limit = 6, available = True, enabled = True):
         if self.collections: kwargs["collections"] = {"$in" : [self.collections[0].id]}
         elif self.categories: kwargs["categories"] = {"$in" : [self.categories[0].id]}
         elif self.colors: kwargs["colors"] = {"$in" : [self.colors[0].id]}
-        elif self.brands: kwargs["brands"] = {"$in" : [self.brands[0].id]}
-        elif self.seasons: kwargs["seasons"] = {"$in" : [self.seasons[0].id]}
+        elif self.brands: kwargs["brands"] = {"$in" : [self.brand.id]}
+        elif self.seasons: kwargs["seasons"] = {"$in" : [self.season.id]}
         kwargs["id"] = {"$nin" : [self.id]}
         kwargs["sort"] = [("id", 1)]
         count = cls.count(**kwargs)