Minor Fix to GUI