Add tests for updating plan.