diff --git a/metastable_baselines/sac/__init__.py b/metastable_baselines/sac/__init__.py
index c0e01b7..5250559 100644
--- a/metastable_baselines/sac/__init__.py
+++ b/metastable_baselines/sac/__init__.py
@@ -1,2 +1,2 @@
-from sb3_trl.trl_sac.policies import CnnPolicy, MlpPolicy, MultiInputPolicy
-from sb3_trl.trl_sac.trl_sac import TRL_SAC
+from ..sac.policies import CnnPolicy, MlpPolicy, MultiInputPolicy
+from ..sac.sac import SAC