Fixed augmentation bugs which interfered with behaviours