Small fix

small fix