Fixed a minor bug in test.