Fixed a possible error in the robustness test.