[FIX] add pyhon import for StringIO