V2 ui fixes