Fix minor build issue due to types mismatch.