Fix build fail