Fix import error in gui.py