Fix the buildding error