fixed the small issue