Fix minor build error