Fix this test to work on 2.0.