Fix StringIO import