diff --git a/src/pygpt_net/core/controller/context.py b/src/pygpt_net/core/controller/context.py
index d113a2e7c..a65eac951 100755
--- a/src/pygpt_net/core/controller/context.py
+++ b/src/pygpt_net/core/controller/context.py
@@ -25,7 +25,7 @@ def __init__(self, window=None):
             'completion': ['chat', 'completion', 'img', 'langchain', 'vision', 'assistant'],
             'img': ['img'],
             'langchain': ['chat', 'completion', 'img', 'langchain', 'vision', 'assistant'],
-            'vision': ['vision'],
+            'vision': ['chat', 'completion', 'img', 'langchain', 'vision', 'assistant'],
             'assistant': ['assistant'],
         }