diff --git a/docs/api/datamodules.rst b/docs/api/datamodules.rst
index 00c0fb998bd..4cc2ff0bd99 100644
--- a/docs/api/datamodules.rst
+++ b/docs/api/datamodules.rst
@@ -6,6 +6,11 @@ torchgeo.datamodules
 Geospatial DataModules
 ----------------------
 
+AgriFieldNet
+^^^^^^^^^^^^
+
+.. autoclass:: AgriFieldNetDataModule
+
 Chesapeake Land Cover
 ^^^^^^^^^^^^^^^^^^^^^
 
diff --git a/tests/conf/agrifieldnet.yaml b/tests/conf/agrifieldnet.yaml
index 603b42d6273..42f2550910a 100644
--- a/tests/conf/agrifieldnet.yaml
+++ b/tests/conf/agrifieldnet.yaml
@@ -15,14 +15,3 @@ data:
     patch_size: 16
   dict_kwargs:
     paths: "tests/data/agrifieldnet"
-
-
-data:
-  class_path: L7IrishDataModule
-  init_args:
-    batch_size: 1
-    patch_size: 32
-    length: 5
-  dict_kwargs:
-    paths: "tests/data/l7irish"
-    download: true
diff --git a/tests/trainers/test_segmentation.py b/tests/trainers/test_segmentation.py
index e829f818bd4..6831d94a649 100644
--- a/tests/trainers/test_segmentation.py
+++ b/tests/trainers/test_segmentation.py
@@ -55,6 +55,7 @@ class TestSemanticSegmentationTask:
     @pytest.mark.parametrize(
         "name",
         [
+            "agrifieldnet",
             "chabud",
             "chesapeake_cvpr_5",
             "chesapeake_cvpr_7",