Edit jUnit test