Add missing changes in gui.py for last commit