diff --git a/aimodel/src/deeplabv3_plus_test.py b/aimodel/src/deeplabv3_plus_test.py
index 7c4a9a5..69e4b9e 100755
--- a/aimodel/src/deeplabv3_plus_test.py
+++ b/aimodel/src/deeplabv3_plus_test.py
@@ -210,7 +210,7 @@ plt.savefig(os.path.join(DIR_OUTPUT, "val_acc.png"))
 
 # Loading the Colormap
 colormap = loadmat(
-    os.path.join(DIR_OUTPUT, "human_colormap.mat")
+    os.path.join(os.path.dirname(DIR_OUTPUT), "human_colormap.mat")
 )["colormap"]
 colormap = colormap * 100
 colormap = colormap.astype(np.uint8)