[minor] Fix broken build