Junit enhancement