Fix build errror