Rebalance 2D fails if superclassification false

Rebalance 2D classes fails with an error if superclassification is set to false:

Traceback (most recent call last):
  File "cryosparc_master/cryosparc_compute/run.py", line 129, in cryosparc_master.cryosparc_compute.run.main
  File "/home/exx/cryosparc/cryosparc_worker/cryosparc_compute/jobs/rebalance_classes_2D/run.py", line 139, in run
    aff_rearr, idxs_by_class = rh.rearrange_pairwise_mtx(affinity_mtx, labels)
UnboundLocalError: local variable 'affinity_mtx' referenced before assignment

This is from a 50 class classification, so this should work I think per the tooltips?