Fix error in the build