Fix remove usage io_interface as context