Small fix 2