diff --git a/l2cs/model.py b/l2cs/model.py
index a7276169..8b433762 100644
--- a/l2cs/model.py
+++ b/l2cs/model.py
@@ -67,7 +67,7 @@ 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
+        return pre_pitch_gaze, pre_yaw_gaze