Fix jUnit test