small fix in compatibility with models