diff --git a/DataStorage/Database/Connection/ConnectionFactory.php b/DataStorage/Database/Connection/ConnectionFactory.php
index e44cfc6b5..b4698cd01 100644
--- a/DataStorage/Database/Connection/ConnectionFactory.php
+++ b/DataStorage/Database/Connection/ConnectionFactory.php
@@ -17,6 +17,8 @@ declare(strict_types=1);
 
 namespace phpOMS\DataStorage\Database\Connection;
 
+use phpOMS\DataStorage\Database\DatabaseType;
+
 /**
  * Database connection factory.
  *
diff --git a/DataStorage/Database/Connection/ConnectionInterface.php b/DataStorage/Database/Connection/ConnectionInterface.php
index 9085ae7ca..a6964f1a7 100644
--- a/DataStorage/Database/Connection/ConnectionInterface.php
+++ b/DataStorage/Database/Connection/ConnectionInterface.php
@@ -51,12 +51,12 @@ interface ConnectionInterface
     /**
      * Get the database type.
      *
-     * @return int
+     * @return string
      *
      * @since  1.0.0
      * @author Dennis Eichhorn <d.eichhorn@oms.com>
      */
-    public function getType() : int;
+    public function getType() : string;
 
     /**
      * Get the database status.