diff --git a/returnn/frontend/_numpy_backend.py b/returnn/frontend/_numpy_backend.py
index 285fa41f3f..d6927a0cf5 100644
--- a/returnn/frontend/_numpy_backend.py
+++ b/returnn/frontend/_numpy_backend.py
@@ -131,6 +131,7 @@ def range_over_dim(dim: Dim, *, dtype: Optional[str] = None, device: Optional[st
         """
         :param dim:
         :param dtype:
+        :param device:
         :return: tensor with shape [dim]
         """
         if not dtype and dim.dyn_size_ext: