JUnit Fixes