Fixing test that was failing with removed ?