Small fix and style