removed the self.dim in group ctrl,
which requires an assumptions that all groups have the same state dim