fixed small code