1 files changed, 7 insertions, 3 deletions
diff --git a/scripts/irbe-cli.py b/scripts/irbe-cli.py
index a72c0664..7e91375e 100755
--- a/scripts/irbe-cli.py
+++ b/scripts/irbe-cli.py
@@ -34,9 +34,13 @@ class cmd_mixin(object):
     self.action = arg
     self.type = "query"
 
-  def client_query_peer_ta(self, arg):
-    """Special handler for --peer_ta option."""
-    self.peer_ta = rpki.x509.X509(Auto_file=arg)
+  def client_query_cms_ta(self, arg):
+    """Special handler for --cms_ta option."""
+    self.cms_ta = rpki.x509.X509(Auto_file=arg)
+
+  def client_query_https_ta(self, arg):
+    """Special handler for --https_ta option."""
+    self.https_ta = rpki.x509.X509(Auto_file=arg)
 
   def client_reply_decode(self):
     pass