diff --git a/sbBrix/__init__.py b/sbBrix/__init__.py
index b3b8f88..d52b3da 100644
--- a/sbBrix/__init__.py
+++ b/sbBrix/__init__.py
@@ -1,7 +1,3 @@
-import os
-
-import numpy as np
-
 from sbBrix.ppo import PPO
 from sbBrix.sac import SAC
 
@@ -9,4 +5,3 @@ __all__ = [
     "PPO",
     "SAC",
 ]
-
diff --git a/sbBrix/sac/__init__.py b/sbBrix/sac/__init__.py
index 813d1fb..59a45df 100644
--- a/sbBrix/sac/__init__.py
+++ b/sbBrix/sac/__init__.py
@@ -1 +1 @@
-from stable_baselines3.sac.sac import SAC
+from sbBrix.sac.sac import SAC