diff --git a/torchgeo/datasets/utils.py b/torchgeo/datasets/utils.py
index 099936ff3b7..0327378d16c 100644
--- a/torchgeo/datasets/utils.py
+++ b/torchgeo/datasets/utils.py
@@ -33,7 +33,7 @@
     "collate_dict",
     "rasterio_loader",
     "dataset_split",
-    "draw_semantic_segmentation_masks"
+    "draw_semantic_segmentation_masks",
 )
 
 
diff --git a/torchgeo/datasets/xview.py b/torchgeo/datasets/xview.py
index fb405165beb..22a5fc85796 100644
--- a/torchgeo/datasets/xview.py
+++ b/torchgeo/datasets/xview.py
@@ -178,7 +178,7 @@ def plot(self, index: int, alpha: float = 0.5) -> plt.Figure:
             index: the index of the sample to plot
 
         Returns:
-
+            the matplotlib.pyplot figure
         """
         sample = self[index]
         image1 = draw_semantic_segmentation_masks(