Fixed build fail