fixing undistribute-remove-after-enum-coll optimizer rule.
fixing distribute-in-cluster tests
fixing undistribute-remove-after-enum-coll tests