Fixed missing reference to model