Fixed tests.py to pass.