fix test for new logic