fix test that was using treatment