diff --git a/backend/experiment/management/commands/templates/test_experiment.py b/backend/experiment/management/commands/templates/test_experiment.py
index 539a41681..7a61e794b 100644
--- a/backend/experiment/management/commands/templates/test_experiment.py
+++ b/backend/experiment/management/commands/templates/test_experiment.py
@@ -1,4 +1,4 @@
-from unittest import TestCase
+from django.test import TestCase
 
 from experiment.models import Experiment
 from participant.models import Participant