diff --git a/apps/ai/main.py b/apps/ai/main.py
index 6ad09b19..9cf04977 100644
--- a/apps/ai/main.py
+++ b/apps/ai/main.py
@@ -7,6 +7,7 @@ myLogger = logging.getLogger('log')
 
 algo_dict = {
     "helmet": "apps.ai.client.helmet",
+    "helmet2": "apps.ai.client.helmet2",
     "fire1": "apps.ai.client.fire1"
 }
 
diff --git a/apps/develop/views.py b/apps/develop/views.py
index f5f1f203..47113eda 100755
--- a/apps/develop/views.py
+++ b/apps/develop/views.py
@@ -214,7 +214,7 @@ class TestViewSet(CustomGenericViewSet):
         """
         code = request.data['code']
         vc = TDevice.objects.get(code=code)
-        snap_and_analyse(vc, ['fire1'], Opl.objects.get(id='1568880208688320512'))
+        snap_and_analyse(vc, ['fire1', 'helmet'], Opl.objects.get(id='1572195171456651264'))
         return Response()
 
     @action(methods=['post'], detail=False, serializer_class=Serializer)