minor bug fix

minor bug fix