diff --git a/alkatorapi/views.py b/alkatorapi/views.py
index b2093d4..9369733 100644
--- a/alkatorapi/views.py
+++ b/alkatorapi/views.py
@@ -12,7 +12,7 @@ import glob
 import PIL.Image
 import random
 
-from .models import User, ALKATOR_CHOICES_DICT
+from .models import User, ALKATOR_CHOICES_DICT, ALKATOR_CLASSES
 from alkator.settings import COMGATE_MERCHANT, COMGATE_SECRET, COMGATE_TEST