diff --git a/src/postcar/__about__.py b/src/postcar/__about__.py
index b527cdd..227be3b 100644
--- a/src/postcar/__about__.py
+++ b/src/postcar/__about__.py
@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present tunnelworks <git@tunnelworks.org>
 #
 # SPDX-License-Identifier: Apache-2.0
-__version__ = "0.0.6"
+__version__ = "0.0.7"
diff --git a/src/postcar/db/migrations/classes.py b/src/postcar/db/migrations/classes.py
index 85ae364..2f7fe0d 100644
--- a/src/postcar/db/migrations/classes.py
+++ b/src/postcar/db/migrations/classes.py
@@ -3,6 +3,7 @@
 
 
 if t.TYPE_CHECKING:
+    from psycopg.abc import Query
     from postcar._types import Connection, Module
 
 
@@ -34,8 +35,8 @@ async def migrate(self) -> None:
     async def rollback(self) -> None:
         raise NotImplementedError()
 
-    def get_forward(self) -> t.Optional[str]:
+    def get_forward(self) -> t.Optional["Query"]:
         return None
 
-    def get_rollback(self) -> t.Optional[str]:
+    def get_rollback(self) -> t.Optional["Query"]:
         return None