Fix wrong set_status call after refactor