Don't override the backwards compat for the param_dims argument, as pointed out by @mhvk