Fix minor build issue