(blindly) fix a couple of unit tests