minor bug fix

oops