Bring JUnit test back