fix ui errors