diff --git a/Model/Command/ProfileInvalidate.php b/Model/Command/ProfileInvalidate.php
index 0d97994..746ba4c 100755
--- a/Model/Command/ProfileInvalidate.php
+++ b/Model/Command/ProfileInvalidate.php
@@ -56,7 +56,7 @@ public function run(InputInterface $input, OutputInterface $output)
             $where['customer_id IN (?)'] = [$input->getArguments()['customer-id']];
         }
         return $connection->update(
-            $this->contentResource->getTable('datatrics_profile'),
+            $this->profileResource->getTable('datatrics_profile'),
             ['status' => 'Queued for Update'],
             $where
         );
diff --git a/Service/ProductData/AttributeCollector/Data/AttributeMapper.php b/Service/ProductData/AttributeCollector/Data/AttributeMapper.php
index a9ece4e..3a8b653 100755
--- a/Service/ProductData/AttributeCollector/Data/AttributeMapper.php
+++ b/Service/ProductData/AttributeCollector/Data/AttributeMapper.php
@@ -76,7 +76,7 @@ class AttributeMapper
     /**
      * Store ID
      *
-     * @var int
+     * @var int|array
      */
     private $storeId;
 
diff --git a/Service/ProductData/AttributeCollector/Data/Category.php b/Service/ProductData/AttributeCollector/Data/Category.php
index ec8bbe0..0997c51 100755
--- a/Service/ProductData/AttributeCollector/Data/Category.php
+++ b/Service/ProductData/AttributeCollector/Data/Category.php
@@ -325,7 +325,7 @@ private function getRootCategoryId()
 
     /**
      * @param array $data
-     * @return mixed
+     * @return array
      * @throws NoSuchEntityException
      */
     private function mergeUrl(array $data): array