small logic fix

oops =p