Fix calls to extension model