Fix unit test that was borken in 6d49adaa965c17a3d9704ed4d85879cd88e5b185