diff --git a/app/Http/Controllers/API/v2/CompanyController.php b/app/Http/Controllers/API/v2/CompanyController.php
index aaeb2ee..94fd907 100644
--- a/app/Http/Controllers/API/v2/CompanyController.php
+++ b/app/Http/Controllers/API/v2/CompanyController.php
@@ -55,9 +55,11 @@ class CompanyController extends Controller
         $companyAPI = new CompanyAPI($fetcher);
 
         $companyMap = $companyAPI->getMap($city_uuid);
-        $companyAPI->saveMap($companyMap);
+        $companyAPI->saveMap($companyMap, $city);
 
-        return new CompanyCollection(Company::all());
+        $companies = $city->getCompanies();
+
+        return new CompanyCollection($companies);
     }
 
     /**