diff --git a/src/main/java/org/testng/internal/TestMethodWorker.java b/src/main/java/org/testng/internal/TestMethodWorker.java
index deca897..d251264 100644
--- a/src/main/java/org/testng/internal/TestMethodWorker.java
+++ b/src/main/java/org/testng/internal/TestMethodWorker.java
@@ -205,7 +205,7 @@
     //
     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) {