Fix TestNG Problem