Fixed another issue

Fixed an issue