Test fix issue

Fixed #1