diff --git a/analysis/load_env_data.py b/analysis/load_env_data.py
index 2136c650..7bab9073 100644
--- a/analysis/load_env_data.py
+++ b/analysis/load_env_data.py
@@ -3,7 +3,7 @@
 """
 
 from relaqs import RESULTS_DIR
-from relaqs.api import load_pickled_env_data
+from relaqs.api.utils import load_pickled_env_data
 
 data_path = RESULTS_DIR + '2024-01-24_11-37-15_X/env_data.pkl'
 
diff --git a/src/relaqs/api/__init__.py b/src/relaqs/api/__init__.py
index 8d1478f1..4e6f1310 100644
--- a/src/relaqs/api/__init__.py
+++ b/src/relaqs/api/__init__.py
@@ -1,4 +1,4 @@
 from .training import TrainRLLib
 from .callbacks import GateSynthesisCallbacks
 from .gates import Gate
-# from .utils import gate_fidelity, dm_fidelity, load_pickled_env_data
+