diff --git a/alwrity.py b/alwrity.py
index a348e64a..9dfc9926 100644
--- a/alwrity.py
+++ b/alwrity.py
@@ -10,6 +10,7 @@ from lib.utils.ui_setup import setup_ui, setup_tabs
 from lib.utils.api_key_manager import check_api_keys, check_llm_environs
 from lib.utils.content_generators import ai_writers, content_planning_tools, blog_from_keyword, story_input_section, essay_writer, ai_news_writer, ai_finance_ta_writer, write_ai_prod_desc, do_web_research, competitor_analysis, ai_agents_content_planner
 from lib.utils.seo_tools import ai_seo_tools
+from lib.utils.alwrity_utils import ai_agents_team, ai_social_writer
 from lib.utils.file_processor import load_image, read_prompts, write_prompts
 from lib.utils.voice_processing import record_voice
 
diff --git a/lib/utils/content_generators.py b/lib/utils/content_generators.py
index 27e639a5..74c42c50 100644
--- a/lib/utils/content_generators.py
+++ b/lib/utils/content_generators.py
@@ -1,7 +1,8 @@
 import streamlit as st
-from lib.utils.alwrity_utils import blog_from_keyword, ai_agents_team, essay_writer, ai_news_writer, ai_seo_tools, ai_finance_ta_writer, ai_social_writer, do_web_research, competitor_analysis, blog_from_audio
+from lib.utils.alwrity_utils import (
     blog_from_keyword, ai_agents_team, essay_writer, ai_news_writer, ai_seo_tools,
-    ai_finance_ta_writer, ai_social_writer, do_web_research, competitor_analysis
+    ai_finance_ta_writer, ai_social_writer, do_web_research, competitor_analysis, blog_from_audio
+)
 )
 from lib.ai_writers.ai_story_writer.story_writer import story_input_section
 from lib.ai_writers.ai_product_description_writer import write_ai_prod_desc