Fixed another problem for the robustness test.