diff --git a/tests/testapk/src/java/com/android/microdroid/test/MicrodroidTests.java b/tests/testapk/src/java/com/android/microdroid/test/MicrodroidTests.java
index c7bb011..14ea402 100644
--- a/tests/testapk/src/java/com/android/microdroid/test/MicrodroidTests.java
+++ b/tests/testapk/src/java/com/android/microdroid/test/MicrodroidTests.java
@@ -142,7 +142,12 @@
         }
 
         void forceStop(VirtualMachine vm) {
-            this.onDied(vm, VirtualMachineCallback.DEATH_REASON_KILLED);
+            try {
+                vm.stop();
+                mExecutorService.shutdown();
+            } catch (VirtualMachineException e) {
+                throw new RuntimeException(e);
+            }
         }
 
         @Override
@@ -160,12 +165,7 @@
         @Override
         @CallSuper
         public void onDied(VirtualMachine vm, @DeathReason int reason) {
-            try {
-                vm.stop();
-                mExecutorService.shutdown();
-            } catch (VirtualMachineException e) {
-                throw new RuntimeException(e);
-            }
+            mExecutorService.shutdown();
         }
     }