1 files changed, 2 insertions, 1 deletions
diff --git a/camera.py b/camera.py
index d9b70c7..52a0add 100755
--- a/camera.py
+++ b/camera.py
@@ -5,6 +5,7 @@ from threading import Thread, Lock
 import os
 import sys
 from color import map_wavelength, map_to_color
+import matplotlib.pyplot as plt
 
 from gpu import GPU, CUDAFuncs
 from tools import timeit
@@ -431,7 +432,7 @@ class EventViewer(Camera):
             t[i] = channel.t
             q[i] = channel.q
 
-        self.gpu.color_solids(solid_ids, map_to_color(t, (t.min(), t.mean())))
+        self.gpu.color_solids(solid_ids, map_to_color(t, range=(t.min(),t.mean())))
         self.update()
 
     def process_event(self, event):