diff --git a/beginner_source/basics/data_tutorial.py b/beginner_source/basics/data_tutorial.py
index 2c46b33122..53d481644b 100644
--- a/beginner_source/basics/data_tutorial.py
+++ b/beginner_source/basics/data_tutorial.py
@@ -234,6 +234,7 @@ def __getitem__(self, idx):
 img = train_features[0].squeeze()
 label = train_labels[0]
 plt.imshow(img, cmap="gray")
+plt.axis("off")
 plt.show()
 print(f"Label: {label}")