change the IO for new framework