Fix build error introduced