diff --git a/fancy_gym/black_box/black_box_wrapper.py b/fancy_gym/black_box/black_box_wrapper.py
index 9d41615..2a2a3f5 100644
--- a/fancy_gym/black_box/black_box_wrapper.py
+++ b/fancy_gym/black_box/black_box_wrapper.py
@@ -2,7 +2,6 @@ from typing import Tuple, Optional, Callable
 
 import gym
 import numpy as np
-import torch
 from gym import spaces
 from mp_pytorch.mp.mp_interfaces import MPInterface