Fix the tests after algorithm adjustment

In 6f208cb938, the difficulty for each stage was changed. This fixes the tests which were dependent.