diff --git a/src/layers/InputLayer.js b/src/layers/InputLayer.js
index 69f1384..017c04e 100644
--- a/src/layers/InputLayer.js
+++ b/src/layers/InputLayer.js
@@ -16,8 +16,8 @@ export default class InputLayer extends Layer {
       shape = []
     } = attrs
 
-    this.shape = attrs.batch_input_shape && attrs.batch_input_shape.length
-      ? attrs.batch_input_shape.slice(1)
+    this.shape = attrs.batchInputShape && attrs.batchInputShape.length
+      ? attrs.batchInputShape.slice(1)
       : shape
   }