diff --git a/src/nebari_plugin_mlflow_aws/__init__.py b/src/nebari_plugin_mlflow_aws/__init__.py
index 7270b6a..7b33cf1 100644
--- a/src/nebari_plugin_mlflow_aws/__init__.py
+++ b/src/nebari_plugin_mlflow_aws/__init__.py
@@ -75,7 +75,7 @@ def _attempt_keycloak_connection(
             time.sleep(timeout)
         return False
     
-    def check(self, stage_outputs: Dict[str, Dict[str, Any]]) -> bool:
+    def check(self, stage_outputs: Dict[str, Dict[str, Any]], disable_prompt=False) -> bool:
                 
         # TODO: Module requires EKS cluster is configured for IRSA.  Once Nebari version with IRSA is released, should update
         # this error message and also minimum Nebari version in pyproject.toml