2 files changed, 7 insertions, 5 deletions
diff --git a/spec/javascripts/vehicle_journeys/actions_spec.js b/spec/javascripts/vehicle_journeys/actions_spec.js
index 219fea96b..5eb3abe45 100644
--- a/spec/javascripts/vehicle_journeys/actions_spec.js
+++ b/spec/javascripts/vehicle_journeys/actions_spec.js
@@ -39,15 +39,16 @@ describe('when toggling arrivals', () => {
 })
 describe('when updating vjas time', () => {
   it('should create an action to update time', () => {
-    const val = 33, subIndex = 0, index = 0, timeUnit = 'minute', isDeparture = true
+    const val = 33, subIndex = 0, index = 0, timeUnit = 'minute', isDeparture = true, isArrivalsToggled = true
     const expectedAction = {
       type: 'UPDATE_TIME',
       val,
       subIndex,
       index,
       timeUnit,
-      isDeparture
+      isDeparture,
+      isArrivalsToggled
     }
-    expect(actions.updateTime(val, subIndex, index, timeUnit, isDeparture)).toEqual(expectedAction)
+    expect(actions.updateTime(val, subIndex, index, timeUnit, isDeparture, isArrivalsToggled)).toEqual(expectedAction)
   })
 })
diff --git a/spec/javascripts/vehicle_journeys/reducers/vehicle_journeys_spec.js b/spec/javascripts/vehicle_journeys/reducers/vehicle_journeys_spec.js
index d6ad1c8bd..ee075a203 100644
--- a/spec/javascripts/vehicle_journeys/reducers/vehicle_journeys_spec.js
+++ b/spec/javascripts/vehicle_journeys/reducers/vehicle_journeys_spec.js
@@ -59,7 +59,7 @@ describe('vehicleJourneys reducer', () => {
   })
 
   it('should handle UPDATE_TIME', () => {
-    const val = 33, subIndex = 0, index = 0, timeUnit = 'minute', isDeparture = true
+    const val = 33, subIndex = 0, index = 0, timeUnit = 'minute', isDeparture = true, isArrivalsToggled = true
     let newVJAS = [{
       arrival_time : "2000-01-01T00:00:00+01:00",
       departure_time : "2000-01-01T00:33:00+01:00",
@@ -73,7 +73,8 @@ describe('vehicleJourneys reducer', () => {
         subIndex,
         index,
         timeUnit,
-        isDeparture
+        isDeparture,
+        isArrivalsToggled
       })
     ).toEqual([newVJ, state[1]])
   })