fix the build issue