fix the import and write a test.