Update example_test.py

make test pass again