fix test for 2.0
fix test