Fix build error caused by 96b499d