Update tests.py

tiny fix