Remove unnecessary import in tests