You need to do a collapse here, as split is not sufficient (as uh resides in the mixed space, with the array being that of the mixed function space: dolfinx.fem — DOLFINx 0.9.0 documentation)
i.e. call uh = wh.split()[0].collapse() or uh = wh.sub(0).collapse()