Fix bamboo compilation error due to unnecessary import in test class