fix for undo error in tests