diff --git a/src/pymongo_migrate/generate.py b/src/pymongo_migrate/generate.py
index e7f5911..439a835 100644
--- a/src/pymongo_migrate/generate.py
+++ b/src/pymongo_migrate/generate.py
@@ -8,16 +8,17 @@
 {description}
 """
 import pymongo
+import pymongo.database
 
 name = {name!r}
 dependencies = {dependencies!r}
 
 
-def upgrade(db: "pymongo.database.Database"):
+def upgrade(db: pymongo.database.Database):
     pass
 
 
-def downgrade(db: "pymongo.database.Database"):
+def downgrade(db: pymongo.database.Database):
     pass
 '''
 MAX_NAME_LEN = 60