Fix up the test_io code

See title