Removing unnecessary import in test file