revert change to unit_test.py