1 files changed, 3 insertions, 6 deletions
diff --git a/tests/unit/DataEntitiesTest.php b/tests/unit/DataEntitiesTest.php
index 350e56d6..42e25faf 100644
--- a/tests/unit/DataEntitiesTest.php
+++ b/tests/unit/DataEntitiesTest.php
@@ -12,7 +12,7 @@ class DataEntitiesTest extends \Codeception\Test\Unit
 {
     protected static $BASE_PATH;
 
-    protected $_container;
+    protected ?\PHPPgAdmin\ContainerUtils $container;
 
     //const BASE_PATH = self::BASE_PATH;
 
@@ -61,11 +61,8 @@ class DataEntitiesTest extends \Codeception\Test\Unit
 
     protected function _before(): void
     {
-        $Helper = $this->getModule('\Helper\Unit');
-        $this->container = $Helper::getContainer();
-        self::$BASE_PATH = self::$BASE_PATH = $this->container->BASE_PATH;
+        $this->container = containerInstance();
+        self::$BASE_PATH = $this->container->BASE_PATH;
         $this->container->get('misc')->setNoDBConnection(true);
-        // Helper
-        // \Codeception\Util\Debug::debug('BASE_PATH is ' . \BASE_PATH);
     }
 }