Fix tests 2.