diff --git a/google/cloud/aiplatform/explain/lit.py b/google/cloud/aiplatform/explain/lit.py
index 432dd9194a..c76840502d 100644
--- a/google/cloud/aiplatform/explain/lit.py
+++ b/google/cloud/aiplatform/explain/lit.py
@@ -206,8 +206,8 @@ def open_lit(
         Raises:
             ImportError if LIT is not installed.
     """
-    widget = notebook.LitWidget(models, datasets, open_in_new_tab=open_in_new_tab)
-    widget.render()
+    widget = notebook.LitWidget(models, datasets)
+    widget.render(open_in_new_tab=open_in_new_tab)
 
 
 def set_up_and_open_lit(