diff --git a/openeogeotrellis/geopysparkdatacube.py b/openeogeotrellis/geopysparkdatacube.py
index 4149e16c..fb46d5a2 100644
--- a/openeogeotrellis/geopysparkdatacube.py
+++ b/openeogeotrellis/geopysparkdatacube.py
@@ -584,7 +584,14 @@ def chunk_polygon(
 
         if isinstance(reducer, dict):
             reducer = GeoPySparkBackendImplementation.accept_process_graph(reducer)
-        chunks: List[Polygon] = chunks.geoms
+
+        if isinstance(chunks, Polygon):
+            chunks = [chunks]
+        elif isinstance(chunks, MultiPolygon):
+            chunks: List[Polygon] = chunks.geoms
+        else:
+            raise ValueError(f"Invalid type for `chunks`: {type(chunks)}")
+
         jvm = get_jvm()
 
         result_collection = None