diff --git a/eICU_synthetic_dataset_generation.py b/eICU_synthetic_dataset_generation.py
index bc14f29..15ef5b1 100644
--- a/eICU_synthetic_dataset_generation.py
+++ b/eICU_synthetic_dataset_generation.py
@@ -131,7 +131,7 @@ def discriminator(x, c, reuse=False):
                 cell=cell,
                 dtype=tf.float32,
                 inputs=decoder_input)
-            rnn_outputs_flat = tf.reshape(rnn_outputs, [-1, hidden_units_g])
+            rnn_outputs_flat = tf.reshape(rnn_outputs, [-1, hidden_units_d])
             logits = tf.matmul(rnn_outputs_flat, W_out_D) + b_out_D
             output = tf.nn.sigmoid(logits)
         return output, logits