Fixed bug in test.