Fixed tests.py