diff --git a/stable_baselines3/common/env_checker.py b/stable_baselines3/common/env_checker.py
index 751b204..6a7f781 100644
--- a/stable_baselines3/common/env_checker.py
+++ b/stable_baselines3/common/env_checker.py
@@ -5,7 +5,7 @@ import gym
 from gym import spaces
 import numpy as np
 
-from stable_baselines.common.vec_env import DummyVecEnv, VecCheckNan
+from stable_baselines3.common.vec_env import DummyVecEnv, VecCheckNan
 
 
 def _enforce_array_obs(observation_space: spaces.Space) -> bool: