(JUnit Tests) Fixing refactored imports.