fixing ui errors
fixing ui errors