Fix for hardcoded model name