Fix omission in previous commit