From 768d5907e6e59d11537611940d9690b6eed74e1c Mon Sep 17 00:00:00 2001
From: Paul Nation <nonhermitian@gmail.com>
Date: Tue, 28 Sep 2021 11:59:48 -0400
Subject: [PATCH] Fix mapping bug

---
 mthree/mitigation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mthree/mitigation.py b/mthree/mitigation.py
index 45e8d983..7293d62c 100644
--- a/mthree/mitigation.py
+++ b/mthree/mitigation.py
@@ -401,7 +401,7 @@ def apply_correction(self, counts, qubits, distance=None,
 
         if isinstance(qubits, dict):
             # If a mapping was given for qubits
-            qubits = list(qubits)
+            qubits = [list(qubits)]
         elif not any(isinstance(qq, (list, tuple, np.ndarray, dict)) for qq in qubits):
             qubits = [qubits]*len(counts)
         else: