Minor change in header test.py