Update tests.py

second change