minor change to switch statement