fixed small error with code