Fixed an idmrg bug.