diff --git a/codeception.yml b/codeception.yml
index 09681d8..d05b65f 100644
--- a/codeception.yml
+++ b/codeception.yml
@@ -18,7 +18,7 @@ paths:
 coverage:
   enabled: true
   include:
-    - src/*.php
+    - src/FondOfSpryker/*.php
 
 modules:
   enabled:
diff --git a/composer.json b/composer.json
index 01350b4..969856f 100644
--- a/composer.json
+++ b/composer.json
@@ -20,7 +20,7 @@
     "spryker/company-business-unit": "^2.0.0"
   },
   "require-dev": {
-    "fond-of-codeception/spryker": "^1.0",
+    "fond-of-codeception/spryker": "^2.0",
     "spryker/code-sniffer": "*",
     "spryker-sdk/phpstan-spryker": "*"
   },
diff --git a/src/FondOfSpryker/Zed/CompanyUnitAddressDefaultShipping/Persistence/CompanyUnitAddressDefaultShippingPersistenceFactory.php b/src/FondOfSpryker/Zed/CompanyUnitAddressDefaultShipping/Persistence/CompanyUnitAddressDefaultShippingPersistenceFactory.php
index 21647fb..fdf46de 100644
--- a/src/FondOfSpryker/Zed/CompanyUnitAddressDefaultShipping/Persistence/CompanyUnitAddressDefaultShippingPersistenceFactory.php
+++ b/src/FondOfSpryker/Zed/CompanyUnitAddressDefaultShipping/Persistence/CompanyUnitAddressDefaultShippingPersistenceFactory.php
@@ -8,8 +8,6 @@
 
 /**
  * @codeCoverageIgnore
- *
- * @method \FondOfSpryker\Zed\CompanyUnitAddressDefaultShipping\Persistence\CompanyUnitAddressDefaultShippingRepositoryInterface getRepository()
  */
 class CompanyUnitAddressDefaultShippingPersistenceFactory extends AbstractPersistenceFactory
 {