Fixed the test.py file bug.