diff --git a/pts/dataset/file_dataset.py b/pts/dataset/file_dataset.py
index c6c96bf..035c067 100644
--- a/pts/dataset/file_dataset.py
+++ b/pts/dataset/file_dataset.py
@@ -84,7 +84,7 @@ class FileDataset(Dataset):
         for path in self.files():
             for line in JsonLinesFile(path):
                 data = self.process(line.content)
-                data["source"] = SourceContext(source=line.span, row=line.span.line)
+                data["source"] = SourceContext(source=line.span.path, row=line.span.line)
                 yield data
 
     def __len__(self):