diff --git a/paasta_tools/kubernetes_tools.py b/paasta_tools/kubernetes_tools.py
index 61ec72dff8..9801f699bf 100644
--- a/paasta_tools/kubernetes_tools.py
+++ b/paasta_tools/kubernetes_tools.py
@@ -3859,14 +3859,13 @@ def mode_to_int(mode: Optional[Union[str, int]]) -> Optional[int]:
 
 
 def update_crds(
-    kube_client: KubeClient,
+    apiextensions: Union[
+        kube_client.ApiextensionsV1Api, kube_client.ApiextensionsV1beta1Api
+    ],
     desired_crds: Collection[
         Union[V1CustomResourceDefinition, V1beta1CustomResourceDefinition]
     ],
     existing_crds: V1CustomResourceDefinitionList,
-    apiextensions: Union[
-        kube_client.ApiextensionsV1Api, kube_client.ApiextensionsV1beta1Api
-    ],
 ) -> bool:
     success = True
     for desired_crd in desired_crds:
diff --git a/paasta_tools/setup_kubernetes_crd.py b/paasta_tools/setup_kubernetes_crd.py
index 440fe75572..932917c2b9 100644
--- a/paasta_tools/setup_kubernetes_crd.py
+++ b/paasta_tools/setup_kubernetes_crd.py
@@ -132,10 +132,9 @@ def setup_kube_crd(
             desired_crds.append(desired_crd)
 
         if update_crds(
-            kube_client=kube_client,
+            apiextensions=apiextension,
             desired_crds=desired_crds,
             existing_crds=existing_crds,
-            apiextensions=apiextension,
         ):
             return True
     return False