Fix model type alignment