Small fix

Small fix