fix broken ut