diff --git a/scripts/tf_cnn_benchmarks/preprocessing.py b/scripts/tf_cnn_benchmarks/preprocessing.py
index 6213e5c1..bfdf08dd 100644
--- a/scripts/tf_cnn_benchmarks/preprocessing.py
+++ b/scripts/tf_cnn_benchmarks/preprocessing.py
@@ -503,7 +503,7 @@ def minibatch(self, dataset, subset, use_datasets, cache_data,
         if not file_names:
           raise ValueError('Found no files in --data_dir matching: {}'
                            .format(glob_pattern))
-        ds = tf.data.TFRecordDataset.list_files(file_names)
+        ds = tf.data.TFRecordDataset.list_files(glob_pattern)
         ds = ds.apply(
             interleave_ops.parallel_interleave(
                 tf.data.TFRecordDataset, cycle_length=10))