From 04d85ac2e2d9999d60df3fd2de2ec4a419c68452 Mon Sep 17 00:00:00 2001
From: Antonin RAFFIN <antonin.raffin@ensta.org>
Date: Tue, 5 May 2020 16:32:08 +0200
Subject: [PATCH] Fix import in tests

---
 tests/test_vec_check_nan.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_vec_check_nan.py b/tests/test_vec_check_nan.py
index 85dd884..a04410a 100644
--- a/tests/test_vec_check_nan.py
+++ b/tests/test_vec_check_nan.py
@@ -2,7 +2,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
 
 
 class NanAndInfEnv(gym.Env):