Fix expected_status types