added missing equal-dims constraints