Fixed small issue

Fixed small issue