Use JUnit asserts for test