Fix compilation error with io module