Fix error in build