diff --git a/autokeras/utils/io_utils.py b/autokeras/utils/io_utils.py
index 7f72f0275..c91f9eed7 100644
--- a/autokeras/utils/io_utils.py
+++ b/autokeras/utils/io_utils.py
@@ -336,7 +336,7 @@ def image_dataset_from_directory(
             Only used if `validation_split` is set.
 
     # Returns
-        A `tf.data.Dataset` object, which yields a tuple `(texts, labels)`,
+        A `tf.data.Dataset` object, which yields a tuple `(images, labels)`,
         where `images` has shape `(batch_size, image_size[0], image_size[1],
         num_channels)` where `labels` has shape `(batch_size,)` and type of
         tf.string.