Fix wrong import, and typing issues.

2 files changed, 2 insertions, 2 deletions

diff --git a/gn3/db/phenotypes.py b/gn3/db/phenotypes.py
index 220284d..4b6a121 100644
--- a/gn3/db/phenotypes.py
+++ b/gn3/db/phenotypes.py
@@ -6,7 +6,7 @@ from dataclasses import dataclass
 
 from MySQLdb.cursors import DictCursor
 
-from wqflask.database import Connection as DBConnection
+from gn3.db_utils import Connection as DBConnection
 
 
 @dataclass(frozen=True)
diff --git a/gn3/db_utils.py b/gn3/db_utils.py
index 7d6a445..2f65a33 100644
--- a/gn3/db_utils.py
+++ b/gn3/db_utils.py
@@ -19,7 +19,7 @@ def parse_db_url(sql_uri: str) -> Tuple:
 # pylint: disable=missing-class-docstring, missing-function-docstring, too-few-public-methods
 class Connection(Protocol):
     """Type Annotation for MySQLdb's connection object"""
-    def cursor(self, *args) -> Any:
+    def cursor(self, *args, **kwargs) -> Any:
         """A cursor in which queries may be performed"""
         ...