Remove use of deprecated method.

freynaud · Sep 4, 2011 · 0a809ad · 0a809ad
1 parent 413eb43
commit 0a809ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/testng/internal/TestMethodWorker.java b/src/main/java/org/testng/internal/TestMethodWorker.java
@@ -205,7 +205,7 @@ protected void invokeAfterClassMethods(ITestClass testClass, IMethodInstance mi)
     //
     List<Object> invokeInstances= Lists.newArrayList();
     ITestNGMethod tm= mi.getMethod();
-    if (m_classMethodMap.removeAndCheckIfLast(tm, mi.getInstances()[0])) {
+    if (m_classMethodMap.removeAndCheckIfLast(tm, mi.getInstance())) {
       Map<ITestClass, Set<Object>> invokedAfterClassMethods
           = m_classMethodMap.getInvokedAfterClassMethods();
       synchronized(invokedAfterClassMethods) {