Update ztest.py