diff --git a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/NeuralNetConfiguration.java b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/NeuralNetConfiguration.java
index 18785f3e9a3..a2fcb059aee 100755
--- a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/NeuralNetConfiguration.java
+++ b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/NeuralNetConfiguration.java
@@ -787,10 +787,13 @@ public ListBuilder hiddenLayerSizes(int...hiddenLayerSizes) {
             return this;
         }
 
+        /**
+         * Build the multi layer network
+         * based on this neural network and
+         * overr ridden parameters
+         * @return the configuration to build
+         */
         public MultiLayerConfiguration build() {
-            if(layerwise.size() != hiddenLayerSizes.length + 1)
-                throw new IllegalStateException("Number of hidden layers mut be equal to hidden layer sizes + 1");
-
             List<NeuralNetConfiguration> list = new ArrayList<>();
             for(int i = 0; i < layerwise.size(); i++) {
                 if(confOverrides.get(i) != null)
diff --git a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/factory/LayerFactories.java b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/factory/LayerFactories.java
index ec6dc927184..feb5645cf7f 100644
--- a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/factory/LayerFactories.java
+++ b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/factory/LayerFactories.java
@@ -22,7 +22,6 @@
 import org.deeplearning4j.nn.api.LayerFactory;
 import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
 import org.deeplearning4j.nn.conf.layers.*;
-import org.deeplearning4j.nn.layers.convolution.subsampling.SubsamplingLayer;
 
 /**
  * Static method for finding which layer factory to use