fixed another test that was broken when UI changed