Forgot to update reference to update_code in tasks.