Yi - fix CI failure