I think it would be more future-proof if we use `*args` and `**kwargs` as argument here and pass them on to `get_origin_batch`. This would allow us to have only one function to maintain.