Small fix of previous commit