fixed the build fail