update_mu now produces the expected dimensions (and thus hopefully the expected results