Fix build error. My bad