diff --git a/src/Statement.php b/src/Statement.php
index f15d1d8..22544a6 100644
--- a/src/Statement.php
+++ b/src/Statement.php
@@ -37,7 +37,7 @@ class Statement extends \Doctrine\DBAL\Statement
      * @param $sql
      * @param Connection $conn
      */
-    public function __construct($sql, Connection $conn)
+    public function __construct($sql, ConnectionInterface $conn)
     {
         // Mysqli executes statement on Statement constructor, so we should retry to reconnect here too
         $attempt = 0;