Fixed broken JUnit test