Remove unused kwarg from _check_state_return