MOdified tests.py