diff --git a/bob/learn/tensorflow/datashuffler/Normalizer.py b/bob/learn/tensorflow/datashuffler/Normalizer.py
index ed65dad1beba9006291e65046d3bd9fff8ed67c1..74e264eff68bbdd3d5c22767d54bb28c07ac8b27 100644
--- a/bob/learn/tensorflow/datashuffler/Normalizer.py
+++ b/bob/learn/tensorflow/datashuffler/Normalizer.py
@@ -25,9 +25,8 @@ class MeanOffset(object):
         self.mean_offset = mean_offset
 
     def __call__(self, x):
-
         for i in range(len(self.mean_offset)):
-            x[:, i, :, :] = x[:, i, :, :] - self.mean_offset[i]
+            x[:, :, i] = x[:, :, i] - self.mean_offset[i]
 
         return x
 
diff --git a/bob/learn/tensorflow/network/VGG16_mod.py b/bob/learn/tensorflow/network/VGG16_mod.py
index d2e7afa89ba02b6e2377da32a65940f60939e12e..d902dc1bd50caa169f42435f871b9463768ef7b4 100644
--- a/bob/learn/tensorflow/network/VGG16_mod.py
+++ b/bob/learn/tensorflow/network/VGG16_mod.py
@@ -53,13 +53,13 @@ class VGG16_mod(SequenceNetwork):
 
                  # Forth convolutional block
                  conv5_1_kernel_size=3,
-                 conv5_1_output=512,
+                 conv5_1_output=256,
 
                  conv5_2_kernel_size=3,
-                 conv5_2_output=512,
+                 conv5_2_output=256,
 
                  conv5_3_kernel_size=3,
-                 conv5_3_output=512,
+                 conv5_3_output=256,
 
                  n_classes=10,