Fixed an issue with the build