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);