diff --git a/predict_pose.py b/predict_pose.py
index 46e3ad7d0b0d0e6246e4a594bb7b3d6818947866..9686aadcd6b51174c2de2886330d978ba2559b12 100644
--- a/predict_pose.py
+++ b/predict_pose.py
@@ -153,7 +153,7 @@ if __name__ == '__main__':
     basePath = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Evaluating/' + class_name
     fps = np.loadtxt(f'Generated_Worlds_/Generated/{class_name}/{class_name}_fps_3d.txt')
 
-    images_ls, labels_ls, mask_ls, choice_ls = data.getAllValData(class_name)
+    images_ls, labels_ls, mask_ls, choice_ls = data.getAllValDataFruits(class_name)
     print(len(images_ls))
 
     for i, img in enumerate(images_ls):