Fix issues after merge

app/database.py

@@ -3,7 +3,7 @@
 from datetime import datetime, timedelta
 
 from app import SESSION
-from app.models import Player, Department, DepartmentStat
+from app.models import Player, Department, DepartmentStat, State
 
 
 def get_latest_professor(state_id, department_type):
@@ -34,7 +34,12 @@ def get_department(session, state_id, department_type):
     ).first()
     if department is None:
         department = Department()
-        department.state_id = state_id
+        state = session.query(State).get(state_id)
+        if not state:
+            state = State()
+            state.id = state_id
+            session.add(state)
+        department.state_id = state.id
         department.department_type = department_type
         session.add(department)
         session.commit()