(maint) Repair tests from last merge-up

I missed a couple of failing tests last merge-up, this fixes that issue.

Signed-off-by: Ken Barber <ken@bob.sh>