Update jUnit test