Update reg_test.py