From 31e3e145a2d98b44c1616eb3fce575c710cf95c0 Mon Sep 17 00:00:00 2001
From: spiglerg <spiglerg@gmail.com>
Date: Fri, 19 Apr 2019 12:51:46 +0200
Subject: [PATCH] tests fix

---
 tests/test_environments.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_environments.py b/tests/test_environments.py
index b62a2e60..4f31fe61 100644
--- a/tests/test_environments.py
+++ b/tests/test_environments.py
@@ -1,10 +1,10 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
+import numpy as np
 
-from flatland.core.env import RailEnv, rail_from_GridTransitionMap_generator
+from flatland.envs.rail_env import RailEnv, rail_from_GridTransitionMap_generator
 from flatland.core.transitions import Grid4Transitions
 from flatland.core.transition_map import GridTransitionMap
-import numpy as np
 
 """Tests for `flatland` package."""
 
-- 
GitLab