Fix import of StringIO