fixed test code error