JUnit Tests update