#2: fixes test cases