diff --git a/src/schnetpack_gschnet/datasets/qm9.py b/src/schnetpack_gschnet/datasets/qm9.py
index 3d2b68c..0f8f4eb 100644
--- a/src/schnetpack_gschnet/datasets/qm9.py
+++ b/src/schnetpack_gschnet/datasets/qm9.py
@@ -91,6 +91,7 @@ def __init__(
         splitting: Optional[SplittingStrategy] = None,
         pin_memory: Optional[bool] = None,
         force_preprocessing: Optional[bool] = False,
+        **kwargs,
     ):
         """
 
@@ -176,6 +177,7 @@ def __init__(
             splitting=splitting,
             pin_memory=pin_memory,
             force_preprocessing=force_preprocessing,
+            **kwargs,
         )
 
         self.remove_uncharacterized = remove_uncharacterized