diff --git a/babyai/levels/bonus_levels.py b/babyai/levels/bonus_levels.py
index aaf2b9ed..2c288b1b 100644
--- a/babyai/levels/bonus_levels.py
+++ b/babyai/levels/bonus_levels.py
@@ -1,5 +1,5 @@
 import gym
-from gym_minigrid.envs import Key, Ball, Box
+from gym_minigrid.minigrid import Key, Ball, Box
 from .verifier import *
 from .levelgen import *