Fix model name issue