fixed the build failure