bug fix

minor change