Signatures disappear if you edit post and resubmit it

I noticed in 2.1 if you edit your own post and resubmit it, the post updates but the signature doesn't appear, whereas it did in 2.0. If you edit your own post but cancel the changes, the signature stays however.
0
Comments
Search for
DiscussionController_AfterCommentBody_Handler
in class.signatures.plugin.php and replace it withBase_AfterCommentBody_Handler
This was fixed, but the version number wasn't updated:
https://github.com/vanilla/addons/pull/125
My themes: pure | minusbaseline - My plugins: CSSedit | HTMLedit | InfiniteScroll | BirthdayModule | [all] - PM me about customizations
VanillaSkins.com - Plugins, Themes and Graphics for Vanillaforums OS
That did it - thanks!