diff --git a/src/rbm/layer.py b/src/rbm/layer.py
index 4b811d2..c7a079e 100644
--- a/src/rbm/layer.py
+++ b/src/rbm/layer.py
@@ -114,7 +114,7 @@ def xor():
 	params.num_gibbs_samples = 3
 
 	# Create layer
-	layer = Layer("Layer_0", (3, 16), params)
+	layer = Layer("Layer_0", (3, 1, 0, 16), params)
 
 	# Init weights
 	layer.init(0.01)