Fix bug with model name