Fixed an issue

Fixed an issue