Update tests.py to pass.