diff --git a/tests-unit/main_test.py b/tests-unit/main_test.py
index 040c1a7cf..836c94995 100644
--- a/tests-unit/main_test.py
+++ b/tests-unit/main_test.py
@@ -11,7 +11,7 @@ def mock_yaml_content():
     return {
         'test_config': {
             'base_path': '~/App/',
-            'model1': 'subfolder1',
+            'checkpoints': 'subfolder1',
         }
     }
 
@@ -53,7 +53,7 @@ def test_load_extra_model_paths_expands_userpath(
     load_extra_path_config(dummy_yaml_file_name)
 
     expected_calls = [
-        ('model1', os.path.join(mock_expanded_home, 'App', 'subfolder1')),
+        ('checkpoints', os.path.join(mock_expanded_home, 'App', 'subfolder1')),
     ]
 
     assert mock_add_model_folder_path.call_count == len(expected_calls)