Fixed error in tests.