Fixed lab2 test case