Specify Dumper correctly
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)