diff --git a/virtinst/osdict.py b/virtinst/osdict.py
index 983f8e83e..3b71ac319 100644
--- a/virtinst/osdict.py
+++ b/virtinst/osdict.py
@@ -263,7 +263,7 @@ class _OSDB(object):
         osobj, treeobj = self._os_loader.get_db().guess_os_from_tree(tree)
         if not osobj:
             return None  # pragma: no cover
-        return osobj.get_short_id(), treeobj
+        return osobj.get_short_id(), _OsTree(treeobj)
 
     def list_os(self):
         """