fixed small code mistake