Fixed minor bug in test.