Fixing a small issue