Minor cleanup.

1 files changed, 6 insertions, 8 deletions

diff --git a/lxml_builder.py b/lxml_builder.py
index 95988e4..16c60f4 100644
--- a/lxml_builder.py
+++ b/lxml_builder.py
@@ -1,5 +1,5 @@
 from lxml import etree
-from BeautifulSoup import TreeBuilder
+from BeautifulSoup import TreeBuilder, Comment
 
 class LXMLBuilder(TreeBuilder):
 
@@ -15,6 +15,9 @@ class LXMLBuilder(TreeBuilder):
         self.parser.feed(markup)
         self.parser.close()
 
+    def close(self):
+        pass
+
     def start(self, name, attrs):
         self.soup.handle_starttag(name, attrs)
 
@@ -26,11 +29,6 @@ class LXMLBuilder(TreeBuilder):
 
     def comment(self, content):
         "Handle comments as Comment objects."
-        self._toStringSubclass(content, Comment)
-
-    def _toStringSubclass(self, text, subclass):
-        """Adds a certain piece of text to the tree as a NavigableString
-        subclass."""
         self.soup.endData()
-        self.data(text)
-        self.soup.endData(subclass)
+        self.soup.handle_data(content)
+        self.soup.endData(Comment)