Small fix up