Fix missing switch