bug fix

fixed a bug