Fixed assertions in tests.