Small fix GUI