Bug fix in test.