diff --git a/assignment3/solution.py b/assignment3/solution.py
index 8c3487f..3eb9d05 100644
--- a/assignment3/solution.py
+++ b/assignment3/solution.py
@@ -224,7 +224,7 @@ def ex2_optimized(directory: str, n_bins: int):
     all_dists = [[] for _ in range(len(methods))]
     selected_distances = [[] for _ in range(len(methods))]
 
-    compare_image = uz_image.imread_gray('../assignment2/data/dataset/object_05_4.png', uz_image.ImageType.float64)
+    compare_image = uz_image.imread('../assignment2/data/dataset/object_05_4.png', uz_image.ImageType.float64)
     h_compare_image = uz_image.get_image_bins_gradient_magnitude_and_angles(compare_image) 
 
     imgs = []
@@ -232,7 +232,7 @@ def ex2_optimized(directory: str, n_bins: int):
 
     for i in range(len(img_names)):
         # Firstly read all images
-        current_image = uz_image.imread_gray(f'{directory}/{img_names[i]}', uz_image.ImageType.float64)
+        current_image = uz_image.imread(f'{directory}/{img_names[i]}', uz_image.ImageType.float64)
         imgs.append(current_image)
         current_image_histogram = uz_image.get_image_bins_gradient_magnitude_and_angles(current_image)
         hists.append(current_image_histogram)