diff --git a/tf-ner-poc/src/main/python/normalizer/normalizer.py b/tf-ner-poc/src/main/python/normalizer/normalizer.py
index e720cc1..bc2c6ac 100644
--- a/tf-ner-poc/src/main/python/normalizer/normalizer.py
+++ b/tf-ner-poc/src/main/python/normalizer/normalizer.py
@@ -97,7 +97,7 @@
 
     encoder_emb_inp = tf.transpose(encoder_emb_inp, perm=[1, 0, 2])
 
-    encoder_cell = tf.nn.rnn_cell.BasicLSTMCell(num_units)
+    encoder_cell = tf.nn.rnn_cell.LSTMCell(num_units)
     initial_state = encoder_cell.zero_state(batch_size_ph, dtype=tf.float32)
 
     encoder_outputs, encoder_state = tf.nn.dynamic_rnn(
@@ -123,7 +123,7 @@
         num_units, attention_states,
         memory_sequence_length=encoder_lengths_ph)
 
-    decoder_cell = tf.nn.rnn_cell.BasicLSTMCell(num_units)
+    decoder_cell = tf.nn.rnn_cell.LSTMCell(num_units)
 
     decoder_cell = tf.contrib.seq2seq.AttentionWrapper(decoder_cell, attention_mechanism,
         attention_layer_size=num_units)