diff --git a/mavion/train.py b/mavion/train.py
index 8f440c023f059bcbd80a67db08be9fb1ad429cdb..6afe6465838757d5069d5240a5ea17c9b300bd4d 100644
--- a/mavion/train.py
+++ b/mavion/train.py
@@ -7,13 +7,11 @@ from rotation import *
 if __name__ == "__main__":
 
     pos_target = np.array([0, 0, -20])
-    psi_target = 45
     eul_target = np.array([0, 90, 45])*deg2rad
     quat_target = eul2quat(eul_target)
     vel_target = np.zeros(3)
     rot_target = np.zeros(3)
 
-
     env = MavionEnv()
     env.target = np.concatenate([pos_target, quat_target, np.zeros(3), np.zeros(3)])
     print(env.target)