fixed small isues