Fixing test for recently removed op