Fixed minor issue in unit test.