Fix test diagnostics