Fixed a issue with first test case