diff --git a/test_agentcore_icons.py b/test_agentcore_icons.py
index dff6fd8b..2cf9e6b1 100644
--- a/test_agentcore_icons.py
+++ b/test_agentcore_icons.py
@@ -8,51 +8,53 @@ Validates that all 11 new Agentcore icons can be imported and used correctly.
 from diagrams import Diagram
 from diagrams.aws.ml import (
     Agentcore,
-    AiAgent, 
-    Runtime,
+    AiAgent,
+    BrowserTool,
+    CodeInterpreter,
+    Evaluations,
     Gateway,
     Identity,
-    CodeInterpreter,
-    Observability,
-    BrowserTool,
     Memory,
-    Evaluations,
-    PolicyEngineAgenticGuardrails
+    Observability,
+    PolicyEngineAgenticGuardrails,
+    Runtime,
 )
 
+
 def test_agentcore_icons():
     """Test all Agentcore icons in a sample diagram"""
-    
+
     with Diagram("AWS Agentcore Bedrock Icons Test", show=False, filename="agentcore_test"):
         # Core Agentcore components
         agentcore = Agentcore("Agentcore")
         ai_agent = AiAgent("AI Agent")
         runtime = Runtime("Runtime")
         gateway = Gateway("Gateway")
-        
+
         # Identity and security
         identity = Identity("Identity")
         policy_engine = PolicyEngineAgenticGuardrails("Policy Engine")
-        
+
         # Tools and capabilities
         code_interpreter = CodeInterpreter("Code Interpreter")
         browser_tool = BrowserTool("Browser Tool")
         memory = Memory("Memory")
-        
+
         # Monitoring and evaluation
         observability = Observability("Observability")
         evaluations = Evaluations("Evaluations")
-        
+
         # Create a flow showing the relationships
         agentcore >> ai_agent >> runtime
         ai_agent >> [code_interpreter, browser_tool, memory]
         gateway >> identity >> policy_engine
         [runtime, code_interpreter, browser_tool] >> observability
         observability >> evaluations
-        
+
     print("✅ Test diagram created successfully!")
     print("📁 Generated file: agentcore_test.png")
     print("🔍 Please verify the icons render correctly in the diagram")
 
+
 if __name__ == "__main__":
     test_agentcore_icons()
diff --git a/test_nova_icons.py b/test_nova_icons.py
index 94a39a4c..f3c62235 100644
--- a/test_nova_icons.py
+++ b/test_nova_icons.py
@@ -8,15 +8,16 @@ Tests Amazon Nova, AWS App Studio, Amazon CodeWhisperer, and AWS Neuron
 from diagrams import Diagram
 from diagrams.aws.ml import AmazonNova, AWSAppStudio, AWSNeuron
 
+
 def test_nova_icons():
     """Test the new Nova and AI service icons"""
-    
+
     with Diagram("New AWS AI/ML Services", show=False, filename="nova_test"):
         # Create instances of the new services
         nova = AmazonNova("Amazon Nova")
         app_studio = AWSAppStudio("AWS App Studio")
         neuron = AWSNeuron("AWS Neuron")
-        
+
         # Create a simple flow
         nova >> app_studio >> neuron
 
@@ -24,5 +25,6 @@ def test_nova_icons():
     print("📁 Generated file: nova_test.png")
     print("🔍 Please verify the icons render correctly in the diagram")
 
+
 if __name__ == "__main__":
     test_nova_icons()