Fix switch test