fix tests.py