diff --git a/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py b/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py
index 04d2b4b0..ce2cca47 100644
--- a/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py
+++ b/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py
@@ -164,17 +164,17 @@ def benchmark_loader(
         # Select the appropriate loader directly
         if loader_type == "CSafeLoader":
             try:
-                from yaml import CSafeLoader as YamlLoader  # type: ignore[attr-defined]
+                from yaml import CSafeLoader as SafeLoader  # type: ignore[attr-defined]
             except ImportError:
-                from yaml import SafeLoader as YamlLoader  # type: ignore[assignment]
+                from yaml import SafeLoader  # type: ignore[assignment]
         else:
-            from yaml import SafeLoader as YamlLoader  # type: ignore[assignment]
+            from yaml import SafeLoader  # type: ignore[assignment]
 
         for _ in range(iterations):
             start_time = time.perf_counter()
             # Parse YAML directly instead of using document_string_to_object
             # to avoid the module-level loader selection
-            yaml.load(template_content, Loader=YamlLoader)
+            yaml.load(template_content, Loader=SafeLoader)
             end_time = time.perf_counter()
             times.append((end_time - start_time) * 1000)  # Convert to milliseconds
 
diff --git a/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py b/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py
index 04d2b4b0..ce2cca47 100644
--- a/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py
+++ b/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py
@@ -164,17 +164,17 @@ def benchmark_loader(
         # Select the appropriate loader directly
         if loader_type == "CSafeLoader":
             try:
-                from yaml import CSafeLoader as YamlLoader  # type: ignore[attr-defined]
+                from yaml import CSafeLoader as SafeLoader  # type: ignore[attr-defined]
             except ImportError:
-                from yaml import SafeLoader as YamlLoader  # type: ignore[assignment]
+                from yaml import SafeLoader  # type: ignore[assignment]
         else:
-            from yaml import SafeLoader as YamlLoader  # type: ignore[assignment]
+            from yaml import SafeLoader  # type: ignore[assignment]
 
         for _ in range(iterations):
             start_time = time.perf_counter()
             # Parse YAML directly instead of using document_string_to_object
             # to avoid the module-level loader selection
-            yaml.load(template_content, Loader=YamlLoader)
+            yaml.load(template_content, Loader=SafeLoader)
             end_time = time.perf_counter()
             times.append((end_time - start_time) * 1000)  # Convert to milliseconds