Fix build error [2]