diff --git a/tests/SqlQueryModuleTest.php b/tests/SqlQueryModuleTest.php
index 15d8d1a..300f2cc 100644
--- a/tests/SqlQueryModuleTest.php
+++ b/tests/SqlQueryModuleTest.php
@@ -8,6 +8,7 @@
 use Aura\Sql\ExtendedPdoInterface;
 use Aura\Sql\PdoInterface;
 use InvalidArgumentException;
+use PDO;
 use PHPUnit\Framework\TestCase;
 use Ray\Di\AbstractModule;
 use Ray\Di\Injector;
diff --git a/tests/SqlQueryTest.php b/tests/SqlQueryTest.php
index 8e93965..49b8c74 100644
--- a/tests/SqlQueryTest.php
+++ b/tests/SqlQueryTest.php
@@ -5,6 +5,7 @@
 namespace Ray\Query;
 
 use Aura\Sql\ExtendedPdo;
+use PDO;
 use PHPUnit\Framework\TestCase;
 
 use function file_get_contents;