diff --git a/vaep/plotting/defaults.py b/vaep/plotting/defaults.py
index ca3d7851f..d6e2b5b81 100644
--- a/vaep/plotting/defaults.py
+++ b/vaep/plotting/defaults.py
@@ -31,7 +31,7 @@
 color_model_mapping['MSIMPUTE_MNAR'] = other_colors[7]
 color_model_mapping['TRKNN'] = other_colors[8]
 color_model_mapping['SVDMETHOD'] = other_colors[9]
-other_colors = sns.color_palette()[10:]
+other_colors = other_colors[10:]
 
 
 def assign_colors(models):
diff --git a/vaep/tests/plotting/test_defaults.py b/vaep/tests/plotting/test_defaults.py
new file mode 100644
index 000000000..086a1df63
--- /dev/null
+++ b/vaep/tests/plotting/test_defaults.py
@@ -0,0 +1,9 @@
+from vaep.plotting.defaults import assign_colors
+
+
+def test_assign_colors():
+    expected = [(0.8392156862745098, 0.15294117647058825, 0.1568627450980392),
+                (0.17254901960784313, 0.6274509803921569, 0.17254901960784313),
+                (0.21044753832183283, 0.6773105080456748, 0.6433941168468681)]
+    assigned = assign_colors(['DAE', 'CF', 'Test'])
+    assert assigned == expected