Fixed augment expand and collapse