Fixed issue with build