Fix error in jUnit test case