diff --git a/src/pystencils/backend/emission/dot_printer.py b/src/pystencils/backend/emission/dot_printer.py
index 51eb62f64bb5e75fb6f7b390504e799572428f04..169febee470bba57158ca9ec217c7ed0ac411dba 100644
--- a/src/pystencils/backend/emission/dot_printer.py
+++ b/src/pystencils/backend/emission/dot_printer.py
@@ -288,29 +288,29 @@ class DotContext:
             dtype = dtype.scalar_type
         match dtype:
             case PsIeeeFloatType(64):
-                return self._dtype_color_map[PsIeeeFloatType(64)]
+                return DotContext._dtype_color_map[PsIeeeFloatType(64)]
             case PsIeeeFloatType(32):
-                return self._dtype_color_map[PsIeeeFloatType(32)]
+                return DotContext._dtype_color_map[PsIeeeFloatType(32)]
             case PsIeeeFloatType(16):
-                return self._dtype_color_map[PsIeeeFloatType(16)]
+                return DotContext._dtype_color_map[PsIeeeFloatType(16)]
             case PsSignedIntegerType(64):
-                return self._dtype_color_map[PsSignedIntegerType(64)]
+                return DotContext._dtype_color_map[PsSignedIntegerType(64)]
             case PsSignedIntegerType(32):
-                return self._dtype_color_map[PsSignedIntegerType(32)]
+                return DotContext._dtype_color_map[PsSignedIntegerType(32)]
             case PsSignedIntegerType(16):
-                return self._dtype_color_map[PsSignedIntegerType(16)]
+                return DotContext._dtype_color_map[PsSignedIntegerType(16)]
             case PsSignedIntegerType(8):
-                return self._dtype_color_map[PsSignedIntegerType(8)]
+                return DotContext._dtype_color_map[PsSignedIntegerType(8)]
             case PsUnsignedIntegerType(64):
-                return self._dtype_color_map[PsUnsignedIntegerType(64)]
+                return DotContext._dtype_color_map[PsUnsignedIntegerType(64)]
             case PsUnsignedIntegerType(32):
-                return self._dtype_color_map[PsUnsignedIntegerType(32)]
+                return DotContext._dtype_color_map[PsUnsignedIntegerType(32)]
             case PsUnsignedIntegerType(16):
-                return self._dtype_color_map[PsUnsignedIntegerType(16)]
+                return DotContext._dtype_color_map[PsUnsignedIntegerType(16)]
             case PsUnsignedIntegerType(8):
-                return self._dtype_color_map[PsUnsignedIntegerType(8)]
+                return DotContext._dtype_color_map[PsUnsignedIntegerType(8)]
             case _:
-                return self._dtype_color_map["other"]
+                return DotContext._dtype_color_map["other"]
 
 
 class DotAstPrinter: