Fixing this test too.