Fix the build error for Yaan

Fix the build error.