Fix StringIO imports