Fixed assertion in tests.