Fix assertion in JUnit test.