[Shobhit] fixing ui