Fixed compilation error.

Broken imports due to model=>models rename