that seems to work for the first test case