fixed test issue

solving issue in #319