Fixed assert in test.