I have a form where I have code in both before and after save and before and after undo. The problem is the BO is still dirty after this code has executed and the data is saved  or undone correctly but the toolstrip is not resetting the buttons because it is seeing the dirty BO I assume and if you close the form I am prompted to save the data. 

Just added save() in after undo and it appears to work