Small fix to get build working