Fix tests 4