Fix to simple test case to make it pass