diff --git a/tests/Provider/CommandProvider.php b/tests/Provider/CommandProvider.php
index 6bc66afcd..4b1254c9c 100644
--- a/tests/Provider/CommandProvider.php
+++ b/tests/Provider/CommandProvider.php
@@ -4,6 +4,7 @@
 
 namespace Yiisoft\Db\Tests\Provider;
 
+use Yiisoft\Db\Command\DataType;
 use Yiisoft\Db\Command\Param;
 use Yiisoft\Db\Expression\Expression;
 use Yiisoft\Db\Query\Query;