fixing demo UI