Fixed a test that broke on 3.x