diff --git a/l2cs/model.py b/l2cs/model.py
index a7276169..2e3a5e1b 100644
--- a/l2cs/model.py
+++ b/l2cs/model.py
@@ -65,9 +65,10 @@ def forward(self, x):
 
         
         # gaze
-        pre_yaw_gaze =  self.fc_yaw_gaze(x)
         pre_pitch_gaze = self.fc_pitch_gaze(x)
-        return pre_yaw_gaze, pre_pitch_gaze
+        pre_yaw_gaze =  self.fc_yaw_gaze(x)
+        
+        return pre_pitch_gaze, pre_yaw_gaze