Fix an testng import statement porblem