diff --git a/binarycpython/utils/distribution_functions.py b/binarycpython/utils/distribution_functions.py
index 67b9a4d0e1e68406eb4e6abb0bc21a382e7ededc..334931a27b1728fb6741d77aebb4eb67e13a18f1 100644
--- a/binarycpython/utils/distribution_functions.py
+++ b/binarycpython/utils/distribution_functions.py
@@ -20,7 +20,6 @@ Tasks:
     - TODO: make an n-part power law that's general enough to fix the three part and the 4 part
 """
 
-import gc
 import math
 import json
 
@@ -1379,11 +1378,11 @@ def build_q_table(options, m, p, verbosity=0):
                 # with a flat distribution up to 1.0.
                 if qs[0] == 1.0:
                     qs[0] = 1.0 - 1e-6
-                    qs[1] = 1
+                    qs.append(1)
                     qdata[qs[0]] = 1
                     qdata[qs[1]] = 1
                 else:
-                    qs[1] = 1.0
+                    qs.append(1)
                     qdata[qs[1]] = qs[0]
 
             # We actually should do the extrapolation now.