diff --git a/cavis-dnn/cavis-dnn-nn/src/main/java/org/deeplearning4j/nn/conf/layers/Convolution1DLayer.java b/cavis-dnn/cavis-dnn-nn/src/main/java/org/deeplearning4j/nn/conf/layers/Convolution1DLayer.java
index 513d684b6..9db0c9d61 100644
--- a/cavis-dnn/cavis-dnn-nn/src/main/java/org/deeplearning4j/nn/conf/layers/Convolution1DLayer.java
+++ b/cavis-dnn/cavis-dnn-nn/src/main/java/org/deeplearning4j/nn/conf/layers/Convolution1DLayer.java
@@ -183,9 +183,9 @@ public class Convolution1DLayer extends ConvolutionLayer {
     return true;
   }
 
-  private static final class Convolution1DLayerBuilderImpl extends ConvolutionLayerBuilder<ConvolutionLayer, Convolution1DLayerBuilderImpl> {
-    public ConvolutionLayer build() {
-      ConvolutionLayer l = new ConvolutionLayer(this);
+  private static final class Convolution1DLayerBuilderImpl extends Convolution1DLayerBuilder<Convolution1DLayer, Convolution1DLayerBuilderImpl> {
+    public Convolution1DLayer build() {
+      Convolution1DLayer l = new Convolution1DLayer(this);
       ConvolutionUtils.validateConvolutionModePadding(l.getConvolutionMode(), l.getPadding());
       ConvolutionUtils.validateCnnKernelStridePadding(
               l.getKernelSize(), l.getStride(), l.getPadding());