Add missing require for StringIO