diff --git a/test-app/runtime/src/main/java/com/tns/Runtime.java b/test-app/runtime/src/main/java/com/tns/Runtime.java
index 521255217..dd3881f06 100644
--- a/test-app/runtime/src/main/java/com/tns/Runtime.java
+++ b/test-app/runtime/src/main/java/com/tns/Runtime.java
@@ -938,7 +938,7 @@ private boolean makeInstanceWeakAndCheckIfAlive(int javaObjectID) {
                 if (instance == null) {
                     // The Java was moved from strong to weak, and then the Java instance was collected.
                     weakInstances.remove(javaObjectID);
-                    weakJavaObjectToID.remove(javaObjectID);
+                    weakJavaObjectToID.remove(ref);
                     return false;
                 } else {
                     return true;