diff --git a/code/expt/expt_lmdk_sel.py b/code/expt/expt_lmdk_sel.py
index 76a17b5..1babad3 100644
--- a/code/expt/expt_lmdk_sel.py
+++ b/code/expt/expt_lmdk_sel.py
@@ -63,7 +63,7 @@ def main(args):
             # Get an option for this part
             if len(opts) > 1:
               res_pt, _ = exp_mech.exponential(hist_pt, opts, exp_mech.score, 1.0, e)
-            elif len(opts) > 0:
+            elif len(opts) == 1:
               res_pt = opts[0]
             # Merge options of all parts
             res[pt_i] = np.sum(res_pt)