diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py
index 7066ba7..c494edb 100755
--- a/datasets/raise/extract_noise.py
+++ b/datasets/raise/extract_noise.py
@@ -54,7 +54,7 @@ def getImageNpArray(imageFileName, computeExtremes, color):
         imageFilePath = f'{imagesFolderPath}/nef/{imageFileName}'
     else:
         imageFilePath = f'{imagesFolderPath}/{imageFileName}'
-    imageNpArray = getColorChannel(imageFilePath)
+    imageNpArray = getColorChannel(imageFilePath, color)
 
     if computeExtremes:
         minColor, maxColor = updateExtremes(imageNpArray, minColor, maxColor)