Update the deprecated use of JUnit