diff --git a/.compute b/.compute
index 1a93820d..c1b3ac9d 100644
--- a/.compute
+++ b/.compute
@@ -10,7 +10,8 @@ wget https://www.dropbox.com/s/wqn5v3wkktw9lmo/install.sh?dl=0 -O install.sh
 sudo sh install.sh
 python3 setup.py develop
 # cp -R ${USER_DIR}/GermanData ../tmp/
-python3 distribute.py --config_path config.json  --data_path /data/ro/shared/data/keithito/LJSpeech-1.1/
+cp -R /data/ro/shared/data/keithito/LJSpeech-1.1/ ../tmp/
+python3 distribute.py --config_path config.json  --data_path ../tmp/LJSpeech-1.1/
 # cp -R ${USER_DIR}/Mozilla_22050 ../tmp/
 # python3 distribute.py --config_path config_tacotron_gst.json  --data_path ../tmp/Mozilla_22050/
 python3 distribute.py --config_path config.json  --data_path /data/rw/home/LibriTTS/train-clean-360
diff --git a/layers/tacotron2.py b/layers/tacotron2.py
index c87ffc78..6e914fd7 100644
--- a/layers/tacotron2.py
+++ b/layers/tacotron2.py
@@ -243,6 +243,8 @@ class Decoder(nn.Module):
 
     def inference(self, inputs):
         memory = self.get_go_frame(inputs)
+        memory = self._update_memory(memory)
+
         self._init_states(inputs, mask=None)
 
         self.attention.init_win_idx()