fix import in unit tests