diff --git a/src/NanoParticleTools/machine_learning/data/utils.py b/src/NanoParticleTools/machine_learning/data/utils.py
index d3c67c3..57a5caf 100644
--- a/src/NanoParticleTools/machine_learning/data/utils.py
+++ b/src/NanoParticleTools/machine_learning/data/utils.py
@@ -5,12 +5,10 @@
 import os
 
 SUNSET_SPECIES_TABLE = {
-    1: sorted(["Yb", "Er", "Xsurfacesix"]),
-    2: sorted(["Yb", "Er"]),
-    3: sorted(["Yb", "Er", "Xsurfacesix", "Tm"]),
-    4: sorted(["Yb", "Er"]),
-    5: sorted(["Yb", "Er", "Nd"]),
-    6: sorted(['Yb', 'Er', "Xsurfacesix", 'Tm', 'Nd', 'Ho', 'Eu', 'Sm', 'Dy'])
+    1: sorted(["Yb", "Er", "Nd"]),
+    2: sorted(["Yb", "Er", "Xsurfacesix"]),
+    3: sorted(["Yb", "Er"]),
+    4: sorted(["Yb", "Er", "Xsurfacesix", "Tm"])
 }