diff --git a/tests-unit/comfy_test/folder_path_test.py b/tests-unit/comfy_test/folder_path_test.py
index 6ee931902..eaef7c127 100644
--- a/tests-unit/comfy_test/folder_path_test.py
+++ b/tests-unit/comfy_test/folder_path_test.py
@@ -166,15 +166,20 @@ def test_base_path_change_clears_old(set_base_dir):
 
 
 def test_models_directory_cli_and_getters(temp_dir):
-    with patch.object(sys, 'argv', ["main.py", "--models-directory", temp_dir]):
-        reload(comfy.cli_args)
-        reload(folder_paths)
+    try:
+        with patch.object(sys, 'argv', ["main.py", "--models-directory", temp_dir]):
+            reload(comfy.cli_args)
+            reload(folder_paths)
 
-    assert folder_paths.get_models_directory() == os.path.abspath(temp_dir)
-    assert folder_paths.models_dir == os.path.abspath(temp_dir)
+        assert folder_paths.get_models_directory() == os.path.abspath(temp_dir)
+        assert folder_paths.models_dir == os.path.abspath(temp_dir)
 
-    with pytest.raises(Exception):
-        comfy.cli_args.is_valid_directory(os.path.join(temp_dir, "non_existent_folder_path"))
+        with pytest.raises(Exception):
+            comfy.cli_args.is_valid_directory(os.path.join(temp_dir, "non_existent_folder_path"))
+    finally:
+        with patch.object(sys, 'argv', ["main.py"]):
+            reload(comfy.cli_args)
+            reload(folder_paths)
 
 
 def test_models_dir_getter_setter():