Fix broken switch signature