Fixing small issue