small fix

small fix