Fix model names in the transition script