Fix the local build error