Fix tests.

#5550