Fixed test case and added one more test