diff --git a/training/deepspeech_training/train.py b/training/deepspeech_training/train.py
index 5d4d7a0d..ded45d2d 100644
--- a/training/deepspeech_training/train.py
+++ b/training/deepspeech_training/train.py
@@ -900,21 +900,21 @@ def do_single_file_inference(input_file_path):
         features = create_overlapping_windows(features).eval(session=session)
         features_len = features_len.eval(session=session)
 
-        logits = outputs['outputs'].eval(feed_dict={
+        probs = outputs['outputs'].eval(feed_dict={
             inputs['input']: features,
             inputs['input_lengths']: features_len,
             inputs['previous_state_c']: previous_state_c,
             inputs['previous_state_h']: previous_state_h,
         }, session=session)
 
-        logits = np.squeeze(logits)
+        probs = np.squeeze(probs)
 
         if FLAGS.scorer_path:
             scorer = Scorer(FLAGS.lm_alpha, FLAGS.lm_beta,
                             FLAGS.scorer_path, Config.alphabet)
         else:
             scorer = None
-        decoded = ctc_beam_search_decoder(logits, Config.alphabet, FLAGS.beam_width,
+        decoded = ctc_beam_search_decoder(probs, Config.alphabet, FLAGS.beam_width,
                                           scorer=scorer, cutoff_prob=FLAGS.cutoff_prob,
                                           cutoff_top_n=FLAGS.cutoff_top_n)
         # Print highest probability result