Fix bug in add_diff_step_1_of_2