Fixed new bug in tests.