fix a few ui problems