Small fix

works