Fix code error on update.py