diff --git a/src/courses/tests/test_views.py b/src/courses/tests/test_views.py
index f2334e8..fa12a21 100644
--- a/src/courses/tests/test_views.py
+++ b/src/courses/tests/test_views.py
@@ -1406,7 +1406,7 @@ def test_create_homework_with_widgets_creates_widgets(self):
 
         self.assertEqual(response.status_code, 302)
 
-        from homeworks.models import Homework
+        from homeworks.models import Homework, HomeworkProgressWidget
 
         homework = Homework.objects.get(title="Homework With Widgets")
         widgets = HomeworkProgressWidget.objects.filter(homework=homework)