Fix io test bug