Little optimization on the updater.py