diff --git a/pom.xml b/pom.xml
index 8f6f748f6b1a1e5cf7cf934094e08ffa9e57494d..60286f7cd9d9c72d166f75e7e526a00293c6866f 100644
--- a/pom.xml
+++ b/pom.xml
@@ -17,7 +17,7 @@
     <!-- .. SE-Libraries .................................................. -->
     <emadl.version>0.2.10-SNAPSHOT</emadl.version>
     <CNNTrain.version>0.3.6-SNAPSHOT</CNNTrain.version>
-    <cnnarch-generator.version>0.0.3-SNAPSHOT</cnnarch-generator.version>
+    <cnnarch-generator.version>0.0.4-SNAPSHOT</cnnarch-generator.version>
     <cnnarch-mxnet-generator.version>0.2.17-SNAPSHOT</cnnarch-mxnet-generator.version>
     <cnnarch-caffe2-generator.version>0.2.13-SNAPSHOT</cnnarch-caffe2-generator.version>
     <cnnarch-gluon-generator.version>0.2.8-SNAPSHOT</cnnarch-gluon-generator.version>
diff --git a/src/test/resources/target_code/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py b/src/test/resources/target_code/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py
index c88e708532b8c9c1039ca7f006a8bf0f19f5e9ee..3af2bd8290ebf99bb9bf0de3e422723e03bf0170 100644
--- a/src/test/resources/target_code/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py
+++ b/src/test/resources/target_code/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py
@@ -136,6 +136,7 @@ class CNNSupervisedTrainer_mnist_mnistClassifier_net:
                 predictions_label = batch.label[0].as_in_context(mx_context)
 
                 with autograd.record():
+                    predictions_ = mx.nd.zeros((10,), ctx=mx_context)
 
                     predictions_ = self._networks[0](image_)
 
@@ -172,6 +173,7 @@ class CNNSupervisedTrainer_mnist_mnistClassifier_net:
                 ]
 
                 if True:
+                    predictions_ = mx.nd.zeros((10,), ctx=mx_context)
 
                     predictions_ = self._networks[0](image_)
 
@@ -192,6 +194,7 @@ class CNNSupervisedTrainer_mnist_mnistClassifier_net:
                 ]
 
                 if True:
+                    predictions_ = mx.nd.zeros((10,), ctx=mx_context)
 
                     predictions_ = self._networks[0](image_)