diff --git a/beginner_source/basics/data_tutorial.py b/beginner_source/basics/data_tutorial.py
index 561e9723fde..d04b8c2bac6 100644
--- a/beginner_source/basics/data_tutorial.py
+++ b/beginner_source/basics/data_tutorial.py
@@ -120,6 +120,7 @@
 
 import os
 import pandas as pd
+from torch.utils.data import Dataset
 from torchvision.io import read_image
 
 class CustomImageDataset(Dataset):