ITTE-1159 Fixed some UI bugs