Update test.py

Revert