Fix problem with test.