Fix import for StringIO