diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index 9e605ce208af1620c49d5734dc585bd0ea7b9cb3..c9ae2aba10d0931aab48b7c31b6f753b0154eb06 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -69,14 +69,11 @@ public class Tensorflow1Interface implements DeepLearningInterface
 	public void loadModel(String modelFolder, String modelSource) throws LoadModelException {
 		model = SavedModelBundle.load(modelFolder, "serve");
 		byte[] byteGraph = model.metaGraphDef();
-        try
-        {
-            sig = MetaGraphDef.parseFrom(byteGraph).getSignatureDefOrThrow("serving_default");
-        }
-        catch (InvalidProtocolBufferException e)
-        {
-            System.out.println("Invalid graph");
-        }
+        try {
+			sig = MetaGraphDef.parseFrom(byteGraph).getSignatureDefOrThrow("serving_default");
+		} catch (com.google.protobuf.InvalidProtocolBufferException e) {
+			throw new LoadModelException();
+		}
 	}
 
 	@Override