fix test for 2.0