Update test.py

Reverted to old commit