diff --git a/beginner_source/basics/data_tutorial.py b/beginner_source/basics/data_tutorial.py
index 561e9723fde..b544d0d494d 100644
--- a/beginner_source/basics/data_tutorial.py
+++ b/beginner_source/basics/data_tutorial.py
@@ -121,6 +121,7 @@
 import os
 import pandas as pd
 from torchvision.io import read_image
+from torch.utils.data import Dataset  # ✅ FIXED
 
 class CustomImageDataset(Dataset):
     def __init__(self, annotations_file, img_dir, transform=None, target_transform=None):