fixed warning in tests.py