diff --git a/tikz/dense-block/dense-block.png b/tikz/dense-block/dense-block.png
index 424b079..c2c9103 100644
Binary files a/tikz/dense-block/dense-block.png and b/tikz/dense-block/dense-block.png differ
diff --git a/tikz/dense-block/dense-block.tex b/tikz/dense-block/dense-block.tex
index 7a02d85..b052c12 100644
--- a/tikz/dense-block/dense-block.tex
+++ b/tikz/dense-block/dense-block.tex
@@ -16,9 +16,9 @@
     \path[arrow] (0, 2) edge node[right, midway] {$256$-d in} (input.north);
 
     % first path
-    \node (conv1) at (0,-1.0) [conv-layer] {$k$ @ $3 \times 3 \times 256$};
+    \node (conv1) at (0,-1.0) [conv-layer] {$k$ @ $3 \times 3$};
     \node (conc1) at (0,-2.7) [conv-layer, dashed] {concatenate};
-    \node (conv2) at (0,-4.4) [conv-layer] {$k$ @ $3 \times 3 \times 256$};
+    \node (conv2) at (0,-4.4) [conv-layer] {$k$ @ $3 \times 3$};
     \node (conc2) at (0,-6.1) [conv-layer, dashed] {concatenate};
 
     \path[arrow] (input.south) edge node[right, midway] {$256$-d in} (conv1.north);