diff --git a/diagnostics/ColorSensorDiagnostics.py b/diagnostics/ColorSensorDiagnostics.py
index 44fae16..99deb8c 100644
--- a/diagnostics/ColorSensorDiagnostics.py
+++ b/diagnostics/ColorSensorDiagnostics.py
@@ -4,7 +4,6 @@ from pybricks.tools import wait, StopWatch
 class ColorSensorDiagnostics:
     def __init__(self, hub, colorsensorclass):
         self.colorsensor = None
-        self.colorsensorclass = colorsensorclass
         self.PORT_MAP = {
             "A": Port.A,
             "B": Port.B,
@@ -13,6 +12,7 @@ class ColorSensorDiagnostics:
             "E": Port.E,
             "F": Port.F,
         }
+        self.colorsensorclass = colorsensorclass
     def initializeColorSensor(self):
         VALID_PORTS = {"A", "B", "C", "D", "E", "F"}
         while True: