diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java
index c645661..1f818e7 100644
--- a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java
@@ -70,6 +70,7 @@
 import org.tensorflow.framework.MetaGraphDef;
 import org.tensorflow.framework.SignatureDef;
 import org.tensorflow.framework.TensorInfo;
+import org.yaml.snakeyaml.Yaml;
 
 /**
  * TODO 
@@ -636,6 +637,7 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
         String modelrunnerPath = getPathFromClass(DeepLearningEngineInterface.class);
         String imglib2Path = getPathFromClass(NativeType.class);
         String gsonPath = getPathFromClass(Gson.class);
+        String snakeYaml = getPathFromClass(Yaml.class);
         String jnaPath = getPathFromClass(com.sun.jna.Library.class);
         String jnaPlatformPath = getPathFromClass(com.sun.jna.platform.FileUtils.class);
         if (modelrunnerPath == null || (modelrunnerPath.endsWith("DeepLearningEngineInterface.class")