diff --git a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java
index fde2170a6..760165b3b 100644
--- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java
+++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java
@@ -1178,7 +1178,7 @@ public class LayerOpValidation extends BaseOpValidation {
         final INDArray res = standardized.mulRowVector(gain).addRowVector(bias);
 
         final INDArray output = Nd4j.zerosLike(res);
-        Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, bias, output, 1));
+        Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, bias, output, true, 1));
 
         assertEquals(res, output);
     }
@@ -1216,7 +1216,7 @@ public class LayerOpValidation extends BaseOpValidation {
         final INDArray res = standardized.mulRowVector(gain);
 
         final INDArray output = Nd4j.zerosLike(res);
-        Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, output, 1));
+        Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, output, true, 1));
 
         assertEquals(res, output);
     }