fix import of KeyModel