diff --git a/small_RNA-seq/small_RNA-seq.snakefile b/small_RNA-seq/small_RNA-seq.snakefile
index 960ac61725511adb0c4eaecbeb9f7efde2f9f54f..b45f3bbf1aecc87cdf844fa5e244cbdc4142f51d 100644
--- a/small_RNA-seq/small_RNA-seq.snakefile
+++ b/small_RNA-seq/small_RNA-seq.snakefile
@@ -3792,10 +3792,23 @@ def make_lib_colour_setter(colour_series_dict):
     return (colour_setter, lib2colour)
 
 
-small_type_series = {"pi" : ["pi"], "mi" : ["mi"], "gene" : ["prot_sisiu", "pseu_sisiu"], "repeat" : ["te_sisiu", "satel_sisiu", "simrep_sisiu"]}
+small_type_series = {
+    "pi" : ["pi"],
+    "mi" : ["mi"],
+    "gene" : ["prot_sisiu", "pseu_sisiu"],
+    "repeat" : ["te_sisiu", "satel_sisiu", "simrep_sisiu"]}
+
+
 def make_small_type_colour_setter(small_type_series):
-    base_colours = sns.color_palette("colorblind", 4)
-    series2base_colour = {"pi" : base_colours[0], "mi" : base_colours[1], "gene" : base_colours[2], "repeat" : base_colours[3]}
+    base_colours = sns.color_palette("colorblind", len(small_type_series))
+    # series2base_colour = {
+    #     "pi" : base_colours[0],
+    #     "mi" : base_colours[1],
+    #     "gene" : base_colours[2],
+    #     "repeat" : base_colours[3]}
+    series2base_colour = {
+        key : base_colours[i]
+        for (i, key) in enumerate(small_type_series.keys())}
     small_type2colour = {}
     for (type_series, small_types) in small_type_series.items():
         nb_colours = len(small_types)
@@ -3816,6 +3829,8 @@ def make_small_type_colour_setter(small_type_series):
 
 
 (small_type_colour_setter, small_type2colour) = make_small_type_colour_setter(small_type_series)
+
+
 (lib_colour_setter, lib2colour) = make_lib_colour_setter(genotype_series)
 #def make_sample_plotter(axis, libs=LIBS, treats=TREATS):
 def make_sample_plotter(axis):