From ddc862c0f19c1ab458ce0e0bbb4e5926ba25b6be Mon Sep 17 00:00:00 2001
From: Thomas Kroes <tkroes81@gmail.com>
Date: Mon, 3 Mar 2025 10:23:39 +0100
Subject: [PATCH] Fix color mapping

---
 src/ScatterplotPlugin.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/ScatterplotPlugin.cpp b/src/ScatterplotPlugin.cpp
index 0dad5b7..e66af4a 100644
--- a/src/ScatterplotPlugin.cpp
+++ b/src/ScatterplotPlugin.cpp
@@ -603,8 +603,10 @@ void ScatterplotPlugin::loadColors(const Dataset<Clusters>& clusters)
     {
         for (size_t i = 0; i < static_cast<size_t>(clusterVec.size()); i++)
         {
-            const auto color = clusterVec[i].getColor();
-            localColors[i] = Vector3f(color.redF(), color.greenF(), color.blueF());
+            const auto& cluster = clusterVec[i];
+            const auto color    = cluster.getColor();
+
+            localColors[cluster.getIndices()[0]] = Vector3f(color.redF(), color.greenF(), color.blueF());
         }
 
     }