Use proper JUnit testing