Fixed issue with build error