diff --git a/test-bot.py b/test-bot.py
index d638f53..21b7996 100644
--- a/test-bot.py
+++ b/test-bot.py
@@ -13,7 +13,12 @@ BOT_TOKEN_SECRET_ARN = 'arn:aws:secretsmanager:us-east-1:495395224548:secret:age
 
 def main():
     prompt = ' '.join(sys.argv[1:]) if len(sys.argv) > 1 else 'Say hello.'
-    session = boto3.Session(profile_name='ai1')
+    # Use env vars if present (export-credentials), fall back to profile
+    import os
+    if os.environ.get('AWS_ACCESS_KEY_ID'):
+        session = boto3.Session()
+    else:
+        session = boto3.Session(profile_name='ai1')
     client = session.client(
         'bedrock-agentcore', region_name='us-east-1',
         config=Config(read_timeout=600, connect_timeout=10)