diff --git a/back/migration.py b/back/migration.py
index 067f95e1..0a1151e0 100644
--- a/back/migration.py
+++ b/back/migration.py
@@ -3,10 +3,10 @@ import asyncio
 import json
 from os import path
 
-from app.db import init_db, Entity, Order, Contract, User, AccessToken
+from app.db import init_db, Entity, Contract, ContractTemplate, ProvisionTemplate, User
 
 
-models = [Entity, Order, Contract, User]
+models = [Entity, Contract, User, ContractTemplate, ProvisionTemplate]
 
 
 async def handle_migration(args):