Add missing switch statement