Small fix to build