diff --git a/pungi/metadata.py b/pungi/metadata.py
index 322bf52f..374f4890 100644
--- a/pungi/metadata.py
+++ b/pungi/metadata.py
@@ -164,9 +164,9 @@ def write_tree_info(compose, arch, variant, timestamp=None):
             if not self.short:
                 self.short = compose.conf["product_short"]
 
-    class LoraxTreeInfo(productmd.TreeInfo):
+    class LoraxTreeInfo(productmd.treeinfo.TreeInfo):
         def clear(self):
-            productmd.TreeInfo.clear(self)
+            super(LoraxTreeInfo, self).clear()
             self.product = LoraxProduct(self)
 
     # images