Fix deprecated keyword argument