fix import for new gernalisation of Model