Fix model name from sample code