fixing a bug where the iterator was automatically moved to the next model after undoing augmentaion (closes #45)