fixed test 2 failure