Fix the build error