Update tests.py

Fixed a little mistake.