diff --git a/docs/changelog.md b/docs/changelog.md
index afd9108c..c0ce4c92 100644
--- a/docs/changelog.md
+++ b/docs/changelog.md
@@ -1,5 +1,6 @@
 ## Unreleased
 
+- Add `sema4ai.fixWrongAgentImport` command
 - Show Output button when there are errors while running action or agent package commands
 - Show datasource configuration status in the tree
 - Add Data Sources to the Packages tree
diff --git a/sema4ai/tests/sema4ai_code_tests/_resources/agent-package/actions/MyActions/action-one/actions.py b/sema4ai/tests/sema4ai_code_tests/_resources/agent-package/actions/MyActions/action-one/actions.py
index 173122f6..757a5dd5 100644
--- a/sema4ai/tests/sema4ai_code_tests/_resources/agent-package/actions/MyActions/action-one/actions.py
+++ b/sema4ai/tests/sema4ai_code_tests/_resources/agent-package/actions/MyActions/action-one/actions.py
@@ -12,6 +12,7 @@
 
 HEADLESS_BROWSER = not os.getenv("HEADLESS_BROWSER")
 
+
 @action
 def get_wikipedia_article_summary(article_url: str) -> str:
     """
@@ -31,10 +32,10 @@ def get_wikipedia_article_summary(article_url: str) -> str:
     page.wait_for_load_state("domcontentloaded")
     page.wait_for_load_state("networkidle")
 
-    paragraphs = page.query_selector_all('.mw-content-ltr>p:not(.mw-empty-elt)')
+    paragraphs = page.query_selector_all(".mw-content-ltr>p:not(.mw-empty-elt)")
     summary = paragraphs[0].inner_text()
 
     # Pretty print for log
     print(summary)
 
-    return summary
\ No newline at end of file
+    return summary