diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java b/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java
index 91a0c92..9426015 100644
--- a/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java
+++ b/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java
@@ -458,11 +458,11 @@ else if (task.status == TaskStatus.FAILED)
 			else if (task.status == TaskStatus.CRASHED)
 				throw new RuntimeException();
 			this.runner.close();
+			this.runner = null;
 			return;
 		} else if (this.interprocessing) {
 			return;
 		}
-		model = null;
 		if (model != null) 
 			model.close();
 		model = null;