diff --git a/myosuite/agents/sb3_job_script.py b/myosuite/agents/sb3_job_script.py
index a2f112e3..db8abe38 100644
--- a/myosuite/agents/sb3_job_script.py
+++ b/myosuite/agents/sb3_job_script.py
@@ -10,6 +10,7 @@
 import os
 import json
 import time as timer
+import myosuite
 from stable_baselines3 import PPO, SAC
 from stable_baselines3.common.callbacks import CheckpointCallback
 from stable_baselines3.common.env_util import make_vec_env