Fixed small issue with set command