diff --git a/src/Command/SetCustomerPrivacyPolicyCommand.php b/src/Command/SetCustomerPrivacyPolicyCommand.php
index 8e2f468f..d4431175 100644
--- a/src/Command/SetCustomerPrivacyPolicyCommand.php
+++ b/src/Command/SetCustomerPrivacyPolicyCommand.php
@@ -7,6 +7,8 @@ use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Dotenv\Dotenv;
+
 use Doctrine\Common\Persistence\ObjectManager;
 
 use App\Entity\Customer;
@@ -28,20 +30,21 @@ class SetCustomerPrivacyPolicyCommand extends Command
     {
         $this->setName('customer:setprivacypolicy')
             ->setDescription('Set customer private policy.')
-            ->addArgument('third_party_policy_id', InputArgument::REQUIRED, 'third_party_policy_id')
-            ->addArgument('mobile_policy_id', InputArgument::REQUIRED, 'mobile_policy_id')
-            ->addArgument('promo_policy_id', InputArgument::REQUIRED, 'promo_policy_id' )
-            ->setHelp('Set customer private policy. Order of ids: third party mobile promo');
+            ->setHelp('Set customer private policy.');
     }
 
     protected function execute(InputInterface $input, OutputInterface $output)
     {
-        $third_party_policy_id = $input->getArgument('third_party_policy_id');
-        $mobile_policy_id = $input->getArgument('mobile_policy_id');
-        $promo_policy_id = $input->getArgument('promo_policy_id');
+        // get the policy ids from .env
+        $dotenv = new Dotenv();
+        $dotenv->loadEnv(__DIR__.'/../../.env');
+
+        $policy_promo_id = $_ENV['POLICY_PROMO'];
+        $policy_third_party_id = $_ENV['POLICY_THIRD_PARTY'];
+        $policy_mobile_id = $_ENV['POLICY_MOBILE'];
 
         // get third party policy
-        $third_party_policy = $this->em->getRepository(PrivacyPolicy::class)->findOneBy(['id' => $third_party_policy_id]);
+        $third_party_policy = $this->em->getRepository(PrivacyPolicy::class)->find($policy_third_party_id);
 
         // get customers on third party
         $third_party_customers = $this->em->getRepository(Customer::class)->findBy(['priv_third_party' => true]);
@@ -51,7 +54,7 @@ class SetCustomerPrivacyPolicyCommand extends Command
         }
 
         // get promo policy
-        $promo_policy = $this->em->getRepository(PrivacyPolicy::class)->findOneBy(['id' => $promo_policy_id]);
+        $promo_policy = $this->em->getRepository(PrivacyPolicy::class)->find($policy_promo_id);
 
         // get customers on promo
         $promo_customers = $this->em->getRepository(Customer::class)->findBy(['priv_promo' => true]);
@@ -63,7 +66,7 @@ class SetCustomerPrivacyPolicyCommand extends Command
         $this->em->flush();
 
         // get mobile policy
-        $mobile_policy = $this->em->getRepository(PrivacyPolicy::class)->findOneBy(['id' => $mobile_policy_id]);
+        $mobile_policy = $this->em->getRepository(PrivacyPolicy::class)->find($policy_mobile_id);
 
         // get mobile sessions
         $mobile_sessions = $this->em->getRepository(MobileSession::class)->findAll();
diff --git a/src/Controller/APIController.php b/src/Controller/APIController.php
index f12d750c..5930d1f9 100644
--- a/src/Controller/APIController.php
+++ b/src/Controller/APIController.php
@@ -1816,7 +1816,7 @@ class APIController extends Controller
         if ($priv_promo)
         {
             // find the promo policy
-            $policy = $em->getRepository(PrivacyPolicy::class)->findOneBy(['id' => $policy_promo_id]);
+            $policy = $em->getRepository(PrivacyPolicy::class)->find($policy_promo_id);
 
             // set policy id
             if ($policy != null)
@@ -1828,7 +1828,7 @@ class APIController extends Controller
         if ($priv_third_party)
         {
             // find the third party policy
-            $policy = $em->getRepository(PrivacyPolicy::class)->findOneBy(['id' => $policy_third_party_id]);
+            $policy = $em->getRepository(PrivacyPolicy::class)->find($policy_third_party_id);
 
             // set policy id
             if ($policy != null)