diff --git a/tests/lib/test_sort_keys.py b/tests/lib/test_sort_keys.py
index 437466c..0afe154 100644
--- a/tests/lib/test_sort_keys.py
+++ b/tests/lib/test_sort_keys.py
@@ -8,7 +8,7 @@
     data = yaml.load(input)
     dump_sorted = yaml.dump(data, default_flow_style=False, sort_keys=True)
     dump_unsorted = yaml.dump(data, default_flow_style=False, sort_keys=False)
-    dump_unsorted = yaml.dump(data, default_flow_style=False, sort_keys=False, Dumper=SafeDumper)
+    dump_unsorted = yaml.dump(data, default_flow_style=False, sort_keys=False, Dumper=yaml.SafeDumper)
     if verbose:
         print("INPUT:")
         print(input)
diff --git a/tests/lib3/test_sort_keys.py b/tests/lib3/test_sort_keys.py
index 1850e88..6c51777 100644
--- a/tests/lib3/test_sort_keys.py
+++ b/tests/lib3/test_sort_keys.py
@@ -8,7 +8,7 @@
     data = yaml.load(input)
     dump_sorted = yaml.dump(data, default_flow_style=False, sort_keys=True)
     dump_unsorted = yaml.dump(data, default_flow_style=False, sort_keys=False)
-    dump_unsorted_safe = yaml.dump(data, default_flow_style=False, sort_keys=False, Dumper=SafeDumper)
+    dump_unsorted_safe = yaml.dump(data, default_flow_style=False, sort_keys=False, Dumper=yaml.SafeDumper)
     if verbose:
         print("INPUT:")
         print(input)