Small fix for build process