Works for the test case