Fixed issue in the GUI caused by 'save_only' function