Fix build failed