     step.feedback_html = feedback_html
     TaskStepActions.loaded(step, stepId, taskId)
 
-    routerStub.forceUpdate(taskComponent, {taskDiv, taskComponent, stepId, taskId, state, router, history, correct_answer, feedback_html})
+    taskTestActions.forceUpdate({taskDiv, taskComponent, stepId, taskId, state, router, history, correct_answer, feedback_html})
 
   _updateStep: (newStepId, {taskDiv, taskComponent, stepId, taskId, state, router, history}) ->
     stepId = newStepId
-    {taskDiv, taskComponent, stepId, taskId, state, router, history}
+
+    taskTestActions.forceUpdate({taskDiv, taskComponent, stepId, taskId, state, router, history})
 
   updateStep: (args...) ->
     Promise.resolve(taskTestActions._updateStep(args...))

        actionsForStep.push('advanceStep')
      actionsForStep
    ).flatten().value()

  completeSteps: (args...) ->
    {taskId} = args[0]

    React.addons.TestUtils.Simulate.change(answerElement, answer)
    TaskStepActions.saved(stepId, {answer_id : answer.id})

    taskTestActions.forceUpdate({taskDiv, taskComponent, stepId, taskId, state, router, history, answer})

  saveMultipleChoice: ({taskDiv, taskComponent, stepId, taskId, state, router, history}) ->
    step = TaskStepStore.get(stepId)
