Fixed ModelTest imports.