fixed test bug