diff --git a/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py b/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py
index f822005a..cb8d10ed 100644
--- a/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py
+++ b/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py
@@ -31,8 +31,10 @@
 from pearl.utils.instantiations.environments.contextual_bandit_uci_environment import (
     SLCBEnvironment,
 )
-
-DATA_PATH: str = "./utils/instantiations/environments/uci_datasets"
+if os.path.exists("../Pearl"):
+    DATA_PATH: str = "pearl/utils/instantiations/environments/uci_datasets"
+else:
+    DATA_PATH: str = "./utils/instantiations/environments/uci_datasets"
 
 """
 Experiment config