diff --git a/PhysicsTools/TensorFlowAOT/test/testAOTTools.py b/PhysicsTools/TensorFlowAOT/test/testAOTTools.py
index 72e43fe65183d..bbedcdca4d415 100644
--- a/PhysicsTools/TensorFlowAOT/test/testAOTTools.py
+++ b/PhysicsTools/TensorFlowAOT/test/testAOTTools.py
@@ -11,6 +11,7 @@
 import tempfile
 import functools
 import unittest
+import platform
 
 
 this_dir = os.path.dirname(os.path.abspath(__file__))
@@ -42,6 +43,8 @@ def test_dev_workflow(self, tmp_dir):
         config_file = os.path.join(m.group(1), "share", "test_models", "simple", "aot_config.yaml")
         self.assertTrue(os.path.exists(config_file))
 
+        arch = "{0}-pc-linux".format(platform.processor())
+
         # run the dev workflow
         # create the test model
         cmd = [
@@ -50,6 +53,7 @@ def test_dev_workflow(self, tmp_dir):
             "-o", tmp_dir,
             "--tool-name", "tfaot-model-test",
             "--dev",
+            "--additional-flags=--target_triple=" + arch
         ]
         run_cmd(cmd)