fix _return check