Clean up imports to fix build warning.