Fixed failing test case