Fixed problem with build