Fixing the two borked tests.