diff --git a/lib/Travelynx/Command/work.pm b/lib/Travelynx/Command/work.pm
index 04dbd2d6bb4f35e20604ec9e08bef6260cdfba4f..727a6948af6dc3133d51f0eb8012e8638a138bdd 100644
--- a/lib/Travelynx/Command/work.pm
+++ b/lib/Travelynx/Command/work.pm
@@ -81,9 +81,10 @@ sub run {
 							'in_transit',
 							'count(*) as count',
 							{
-								user_id            => $uid,
-								train_no           => $train->train_no,
-								checkin_station_id => $dep
+								user_id             => $uid,
+								train_no            => $train->train_no,
+								checkin_station_id  => $dep,
+								checkout_station_id => $arr,
 							}
 						)->hash->{count}
 					  )
@@ -94,9 +95,10 @@ sub run {
 								cancelled => 1,
 							},
 							{
-								user_id            => $uid,
-								train_no           => $train->train_no,
-								checkin_station_id => $dep
+								user_id             => $uid,
+								train_no            => $train->train_no,
+								checkin_station_id  => $dep,
+								checkout_station_id => $arr,
 							}
 						);