Fix issue with LanguageModel