Re-applied fix to IO test