fixed error in test.py