Trying to fix custom GUI