Little fixes in update.py