Fix build error caused by in 96b499d