fix small bug on model switch