diff --git a/update-attempt-ids.py b/update-attempt-ids.py
index 728c656772ee9927adf06f7fb1e779b783add57b..66191de9f1cc510296cf817ae6f3a0711e594a01 100644
--- a/update-attempt-ids.py
+++ b/update-attempt-ids.py
@@ -1,11 +1,11 @@
 # -*- coding: utf-8 -*-
 from __future__ import unicode_literals
 
-from django.db import models, migrations
-from course.models import GradeChange
+import django
 
+django.setup()
 
-GradeChange = apps.get_model("course", "GradeChange")
+from course.models import GradeChange
 
 for gchange in GradeChange.objects.all():
     if gchange.flow_session is not None: