fix reference to model to use model now.