Fixed wrong class assignment in ui