Fixed a bug with the test case.