Fix UFS build error