Fix compile errors in unit test.
Remove unused imports.