Fix loading GUI behaviour