Fixed an issue with build