Small fix-up