Switched to JUnit 4 Assert class