Removed case, put if/else to get rid of error