diff --git a/.coverage b/.coverage
index 5710bb4..0c9f6db 100644
Binary files a/.coverage and b/.coverage differ
diff --git a/orm/models.py b/orm/models.py
index 19224aa..096415e 100644
--- a/orm/models.py
+++ b/orm/models.py
@@ -1,7 +1,5 @@
-from __future__ import annotations
-
 import json
-from typing import Any
+from typing import Any, Type
 from typing import Set, Dict
 
 import pydantic
@@ -179,7 +177,7 @@ class Model(metaclass=ModelMetaclass):
         result = await self.__database__.execute(expr)
         return result
 
-    async def load(self) -> Model:
+    async def load(self) -> 'Model':
         expr = self.__table__.select().where(self.pk_column == self.pk)
         row = await self.__database__.fetch_one(expr)
         self.from_dict(dict(row))