fixing a small issue