Restore expected behavior in add/remove level tests