fixed unit-test.py