Fix model names in test