Correct test.py