Remove default case to use the normal function return