diff --git a/models.py b/models.py
index 63e861f..7ec3801 100644
--- a/models.py
+++ b/models.py
@@ -1,4 +1,5 @@
 from config import conf
+from peewee import MySQLDatabase
 from peewee import SqliteDatabase
 from peewee import Model
 from peewee import AutoField, DateField, IntegerField