[Build] Fix error in build