Small fix in previous commit