diff --git a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/validation/OpValidation.java b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/validation/OpValidation.java
index e9ad61c04..3e329ad13 100644
--- a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/validation/OpValidation.java
+++ b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/validation/OpValidation.java
@@ -341,8 +341,8 @@ public class OpValidation {
 
 
         //Finally: check execution/output
-        Map<String,INDArray> outOrig = original.execAll(tc.placeholderValues());
-        Map<String,INDArray> outDe = deserialized.execAll(tc.placeholderValues());
+        Map<String,INDArray> outOrig = original.outputAll(tc.placeholderValues());
+        Map<String,INDArray> outDe = deserialized.outputAll(tc.placeholderValues());
         Preconditions.checkState(outOrig.keySet().equals(outDe.keySet()), "Keysets for execution after deserialization does not match key set for original model");
 
         for(String s : outOrig.keySet()){
diff --git a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/imports/converters/DifferentialFunctionClassHolder.java b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/imports/converters/DifferentialFunctionClassHolder.java
index a92066d7a..82bfdc843 100644
--- a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/imports/converters/DifferentialFunctionClassHolder.java
+++ b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/imports/converters/DifferentialFunctionClassHolder.java
@@ -64,6 +64,7 @@ public class DifferentialFunctionClassHolder {
         add("outputVariables");
         add("tArguments");
         add("iArguments");
+        add("bArguments");
         add("hash");
         add("opName");
         add("sameDiff");