fixed small variable issue