tests.py modified