fixed small problems