diff --git a/src/napari_chatgpt/utils/python/test/modify_code_test.py b/src/napari_chatgpt/utils/python/test/modify_code_test.py
index fa96cf4..c60de62 100644
--- a/src/napari_chatgpt/utils/python/test/modify_code_test.py
+++ b/src/napari_chatgpt/utils/python/test/modify_code_test.py
@@ -25,7 +25,7 @@ def test_modify_code():
 
     assert len(modified_code) >= len(___generated_python_code)
 
-    assert 'multichannel' in modified_code or 'multi-channel' in modified_code
+    assert 'multichannel' in modified_code or 'multi-channel' in modified_code or 'channel' in modified_code