Fixed a bug in our test cases