From 36f72e95608a1e230e402c96cbba432b0a1fb8f0 Mon Sep 17 00:00:00 2001
From: Daniel Farrell <dfarrell@wmci.com>
Date: Thu, 26 Jul 2012 12:06:43 -0400
Subject: [PATCH] Better undo

Removes itself better, returns this to be chainable
---
 jquery.tablescroll.js | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/jquery.tablescroll.js b/jquery.tablescroll.js
index 7596b3f..d526d04 100644
--- a/jquery.tablescroll.js
+++ b/jquery.tablescroll.js
@@ -50,15 +50,16 @@ OTHER DEALINGS IN THE SOFTWARE.
 	{
 		if (options == 'undo')
 		{
-			var container = $(this).parent().parent();
-			if (container.hasClass('tablescroll_wrapper')) 
+			var container = this.parent().parent();
+			if (container.hasClass('tablescroll'))
 			{
 				container.find('.tablescroll_head thead').prependTo(this);
 				container.find('.tablescroll_foot tfoot').appendTo(this);
 				container.before(this);
-				container.empty();
+				container.remove();
+				this.removeClass('tablescroll_body');
 			}
-			return;
+			return this;
 		}
 
 		var settings = $.extend({},$.fn.tableScroll.defaults,options);