diff --git a/src/main/java/plugins/danyfel80/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/plugins/danyfel80/tensorflow/v1/Tensorflow1Interface.java
index c9b182a39ff279ae29c8d9d543b3fea8754d8bc0..355e819929c08f1050c906560302c72c7a3c33a9 100644
--- a/src/main/java/plugins/danyfel80/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/plugins/danyfel80/tensorflow/v1/Tensorflow1Interface.java
@@ -132,9 +132,10 @@ public class Tensorflow1Interface implements DeepLearningInterface
 	@Override
 	public void closeModel() {
 		sig = null;
-		model.session().close();
-		if (model != null)
+		if (model != null) {
+			model.session().close();
 			model.close();
+		}
 		model = null;
 		
 	}