fixed missing imports in models