Fix build error in previous commit