Updated deprecated use of JUnit