Fix omission for warning