Fixed small issue