Fix issue with code not building