Fix tests

Yes.