Small fix to prevent build errors