diff --git a/Dockerfile_b7s b/Dockerfile_b7s
index acf35d8..4f00402 100644
--- a/Dockerfile_b7s
+++ b/Dockerfile_b7s
@@ -1,5 +1,4 @@
-FROM --platform=linux/amd64 696230526504.dkr.ecr.us-east-1.amazonaws.com/allora-inference-base:latest
-# FROM --platform=linux/amd64 allora-inference-base:dev
+FROM --platform=linux/amd64 alloranetwork/allora-inference-base:latest
 
 USER root
 RUN pip install requests
diff --git a/main.py b/main.py
index 4a6b890..4702eb9 100644
--- a/main.py
+++ b/main.py
@@ -15,8 +15,10 @@ def process(token_name):
 if __name__ == "__main__":
     # Your code logic with the parsed argument goes here
     try:
-        if len(sys.argv) >= 2:
-            token_name = sys.argv[1]
+        if len(sys.argv) >= 3:
+            # Not using to discriminate by topicId for simplicity.
+            # topic_id = sys.argv[1]
+            token_name = sys.argv[2]
         else:
             token_name = "ETH"
         process(token_name=token_name)