Fixed problem in tests.