a couple of ui fixes