JUnit tests update