Fixed wrong type in switch model