Fixed the build error