diff --git a/tutorcodejail/templates/codejail/apps/config/tutor.py b/tutorcodejail/templates/codejail/apps/config/tutor.py
index 6f7c557..3722086 100644
--- a/tutorcodejail/templates/codejail/apps/config/tutor.py
+++ b/tutorcodejail/templates/codejail/apps/config/tutor.py
@@ -5,6 +5,7 @@
 class DevelopmentConfig(DevelopmentConfig):
     """Class to use for development context that inherits from DevelopmentConfig."""
 
+    CODE_JAIL = DevelopmentConfig.CODE_JAIL
     {{patch("codejail-common-settings")}}
     {{patch("codejail-development-settings")}}
 
@@ -12,5 +13,6 @@ class DevelopmentConfig(DevelopmentConfig):
 class ProductionConfig(ProductionConfig):
     """Class to use for production context that inherits from ProductionConfig."""
 
+    CODE_JAIL = ProductionConfig.CODE_JAIL
     {{patch("codejail-common-settings")}}
     {{patch("codejail-production-settings")}}