diff --git a/npp.py b/npp.py
index d3771c1..5a50a1e 100644
--- a/npp.py
+++ b/npp.py
@@ -70,6 +70,6 @@ def run_on_qpu(Q, sampler):
 
     ## ------- Return results to user -------
     for sample in sample_set:
-        S1 = [S[i] for i in sample if sample[i] == 1]
-        S0 = [S[i] for i in sample if sample[i] == 0]
+        S1 = [S[i] for i in range(1,8) if sample[i] == 1]
+        S0 = [S[i] for i in range(1,8) if sample[i] == 0]
         print("S0 Sum: ", sum(S0), "\tS1 Sum: ", sum(S1), "\t", S0)