From 503f8dddada48c91f5c466665c9f92382d1c2a32 Mon Sep 17 00:00:00 2001
From: Diatrex <diahshalabi@hotmail.com>
Date: Fri, 4 Sep 2020 16:43:19 +0300
Subject: [PATCH] #2070 emlaktown pyro add

---
 .../visiosoft/cats-module/src/Category/CategoryCriteria.php  | 5 +----
 .../visiosoft/location-module/src/City/CityCriteria.php      | 4 +---
 2 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php b/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php
index b4712eb7b..66c1390d1 100644
--- a/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php
+++ b/addons/default/visiosoft/cats-module/src/Category/CategoryCriteria.php
@@ -20,10 +20,7 @@ class CategoryCriteria extends EntryCriteria
     }
 
     public function getMainCats() {
-        $mainCats = $this->categoryRepository->newQuery()
-            ->whereNull('parent_category_id')
-            ->orderBy('sort_order')
-            ->get();
+        $mainCats = $this->categoryRepository->mainCats();
 
         foreach ($mainCats as $cat) {
             $subCount = $this->categoryRepository->newQuery()->where('parent_category_id', $cat->id)->count();
diff --git a/addons/default/visiosoft/location-module/src/City/CityCriteria.php b/addons/default/visiosoft/location-module/src/City/CityCriteria.php
index d49bd26a6..ee57d86bd 100644
--- a/addons/default/visiosoft/location-module/src/City/CityCriteria.php
+++ b/addons/default/visiosoft/location-module/src/City/CityCriteria.php
@@ -5,8 +5,6 @@ use Anomaly\Streams\Platform\Entry\EntryCriteria;
 class CityCriteria extends EntryCriteria
 {
     public function getSubCities($city) {
-        $this->query->where('parent_country_id', $city)->get();
-
-        return $this;
+        return $this->query->where('parent_country_id', $city)->get();
     }
 }