diff --git a/src/propeller/gp.cljc b/src/propeller/gp.cljc
index 42fd861..f7bc440 100644
--- a/src/propeller/gp.cljc
+++ b/src/propeller/gp.cljc
@@ -36,7 +36,8 @@
        :behavioral-diversity  (float (/ (count (distinct (map :behaviors pop))) (count pop)))
        :average-genome-length (float (/ (reduce + (map count (map :plushy pop))) (count pop)))
        :average-total-error   (float (/ (reduce + (map :total-error pop)) (count pop)))}
-      (if (> (or (:bmx (:variation argmap)) 0) 0) ; using bmx
+      (if (or (> (or (:bmx (:variation argmap)) 0) 0)       ; using bmx
+              (> (or (:bmx-umad (:variation argmap)) 0) 0)) ; using bmx-umad
         {:best-gene-count     (utils/count-genes (:plushy best))
          :average-gene-count  (float (/ (reduce + (map utils/count-genes (map :plushy pop)))
                                         (count pop)))}