Adapt Test utils to restricted IO