Actually pass init_args and init_kwargs through.

I think something like this is what you meant.