diff --git a/godot_rl/wrappers/onnx/stable_baselines_export.py b/godot_rl/wrappers/onnx/stable_baselines_export.py
index 348f8c4e..1752842f 100644
--- a/godot_rl/wrappers/onnx/stable_baselines_export.py
+++ b/godot_rl/wrappers/onnx/stable_baselines_export.py
@@ -1,5 +1,5 @@
 import torch
-from gymnasium.vector.utils import spaces
+from gymnasium import spaces
 from stable_baselines3 import PPO, SAC