diff --git a/flanabot/main.py b/flanabot/main.py
index 6d49fbf..1932330 100644
--- a/flanabot/main.py
+++ b/flanabot/main.py
@@ -1,15 +1,14 @@
+import asyncio
 import os
 
 import flanautils
 
-os.environ |= flanautils.find_environment_variables('../.env')
-
-import asyncio
 from flanabot.bots.flana_disc_bot import FlanaDiscBot
 from flanabot.bots.flana_tele_bot import FlanaTeleBot
 
 
 async def main():
+    os.environ |= flanautils.find_environment_variables('../.env')
     flana_disc_bot = FlanaDiscBot()
     flana_tele_bot = FlanaTeleBot()
 
diff --git a/flanabot/models/punishment.py b/flanabot/models/punishment.py
index 3566815..c7f8df4 100644
--- a/flanabot/models/punishment.py
+++ b/flanabot/models/punishment.py
@@ -4,7 +4,7 @@ import datetime
 from dataclasses import dataclass
 from typing import Any, Callable
 
-from multibot.models import Platform, PunishmentBase, db
+from multibot.models import Platform, PunishmentBase
 
 from flanabot import constants
 from flanabot.models.message import Message
@@ -12,7 +12,7 @@ from flanabot.models.message import Message
 
 @dataclass(eq=False)
 class Punishment(PunishmentBase):
-    collection = db.punishment
+    collection_name = 'punishment'
 
     level: int = 0