Fix GUI after the small API change