diff --git a/linopy/variables.py b/linopy/variables.py
index 0eed704d..1d07e8a1 100644
--- a/linopy/variables.py
+++ b/linopy/variables.py
@@ -1122,7 +1122,7 @@ def to_polars(self) -> pl.DataFrame:
         check_has_nulls_polars(df, name=f"{self.type} {self.name}")
         return df
 
-    def sum(self, dim: str | None = None, **kwargs: Any) -> LinearExpression:
+    def sum(self, dim: DimsLike | None = None, **kwargs: Any) -> LinearExpression:
         """
         Sum the variables over all or a subset of dimensions.