Fixed error in test.