diff --git a/qubesadmin/app.py b/qubesadmin/app.py index ffeb287..e5f2244 100644 --- a/qubesadmin/app.py +++ b/qubesadmin/app.py @@ -216,10 +216,8 @@ class QubesBase(qubesadmin.base.PropertyHolder): pass # then search for index - if label.isdigit(): - for i in self.labels: - if i.index == int(label): - return i + if type(label) == int or label.isdigit(): + return self.labels.values[int(label)] raise KeyError(label)