Clean up and fix unit tests.

- Unit test needs to have test in the name.
- Remove unnecessary imports.