Switch statement fix.