diff --git a/scripts/hyperparameter_tuning/hyperparameter_tuning.py b/scripts/hyperparameter_tuning/hyperparameter_tuning.py
index beca9d61..ab07b572 100644
--- a/scripts/hyperparameter_tuning/hyperparameter_tuning.py
+++ b/scripts/hyperparameter_tuning/hyperparameter_tuning.py
@@ -3,7 +3,7 @@
 from ray import tune
 from ray.air import RunConfig
 from ray.rllib.algorithms.ddpg import DDPGConfig
-from relaqs.environments.gate_synth_env_rllib_Haar import TwoQubitGateSynth, GateSynthEnvRLlibHaarNoisy
+from relaqs.environments import SingleQubitEnv, NoisySingleQubitEnv
 from ray.tune.search.optuna import OptunaSearch
 from relaqs import RESULTS_DIR
 import datetime
@@ -84,7 +84,7 @@ def objective(config):
     return results
 
 if __name__ == "__main__":
-    environment = GateSynthEnvRLlibHaarNoisy
+    environment = NoisySingleQubitEnv
     n_configurations = 25
     n_training_iterations = 100
     save = True
diff --git a/src/relaqs/environments/__init__.py b/src/relaqs/environments/__init__.py
index e69de29b..11b93ba1 100644
--- a/src/relaqs/environments/__init__.py
+++ b/src/relaqs/environments/__init__.py
@@ -0,0 +1,2 @@
+from .single_qubit_env import SingleQubitEnv
+from .noisy_single_qubit_env import NoisySingleQubitEnv