Corrected first test case