diff --git a/datasets/raise/extract_4_color_channel_images_from_raw.py b/datasets/raise/extract_4_color_channel_images_from_raw.py
index 74ee852..9d455ad 100644
--- a/datasets/raise/extract_4_color_channel_images_from_raw.py
+++ b/datasets/raise/extract_4_color_channel_images_from_raw.py
@@ -11,8 +11,11 @@ INTERESTING_POSITION = np.array(ORIGINAL_INTERESTING_POSITION) // 2
 def getImageFilePath(imageIndex):
     return f'../rafael/240424/photos/DSC0{imageIndex}.ARW'
 
-imageFilePath = getImageFilePath(IMAGE_INDEX)
-colorChannel = getColorChannel(imageFilePath, Color.RED)
+def getImageColorChannel(imageIndex, color):
+    imageFilePath = getImageFilePath(imageIndex)
+    return getColorChannel(imageFilePath, Color.RED)
+
+colorChannel = getImageColorChannel(IMAGE_INDEX, Color.RED)
 print(colorChannel.shape)
 
 def crop(image, interestingPosition, yRange, xRange):