Fix test_suite_2.py