diff --git a/code/data.py b/code/data.py
index 618d6801b550ae97119c73796110677e491fd034..235cae1b79246ff4460c181c545502637db93531 100644
--- a/code/data.py
+++ b/code/data.py
@@ -122,12 +122,12 @@ class TestDataModule(pl.LightningDataModule):
             # Load the pyramid/00 per file
 
             #For JSC Training.
-            #pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
-            #cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
+            pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
+            cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
 
             #For Local Machine Training.
-            pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
-            cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
+            #pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
+            #cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
 
             pli_files_list = [file for file in os.listdir(pli_path) if
                               file.endswith(('.h5', '.hdf', '.h4', '.hdf4', '.he2', '.hdf5', '.he5'))]