Fix build error!

My mistake