diff --git a/test/test.py b/test/test.py
index b7579ad..0aa4c14 100644
--- a/test/test.py
+++ b/test/test.py
@@ -1,9 +1,11 @@
+import os
+os.chdir("..")
 from utils.one_hot_encoder import HotEncoder
 import  numpy as np
 import torch
 import skimage
 from torch.nn.functional import one_hot
-img__= skimage.io.imread("/Users/madhav/DataSets/data_semantics/training/semantic_rgb/000035_10.tif")
+img__= skimage.io.imread("/Users/madhav/DataSets/data_semantics/training/semantic_rgb/000034_10.tif")
 encoder= HotEncoder(dir= '/Users/madhav/DataSets/data_semantics/training/semantic_rgb', extension="tif", is_binary= False, n_classes= 30)
 color_dict= encoder.gen_colors()
 onehottt= encoder.PerPixelClassMatrix(img__)