diff --git a/src/utils/image.py b/src/utils/image.py
index 088b0ff..0f93969 100644
--- a/src/utils/image.py
+++ b/src/utils/image.py
@@ -46,8 +46,8 @@ def get_pseudo_rgb(arr: np.ndarray, gamma: float = 0.3) -> np.ndarray:
     a3[a3 > p9999] = p9999
     a3 /= a3.max()
 
-    return np.stack([a1, np.zeros(a1.shape), np.zeros(a1.shape)], axis=0)
-    # return np.stack([a1, a2, a3], axis=0)
+    # return np.stack([a1, np.zeros(a1.shape), np.zeros(a1.shape)], axis=0)
+    return np.stack([a1, a2, a3], axis=0)
 
 
 class ImageLoadError(Exception):