fixed small problem