1 files changed, 4 insertions, 3 deletions
diff --git a/lldb/packages/Python/lldbsuite/test/benchmarks/disassembly/TestDoAttachThenDisassembly.py b/lldb/packages/Python/lldbsuite/test/benchmarks/disassembly/TestDoAttachThenDisassembly.py
index a7c17bcd346..5d759399c41 100644
--- a/lldb/packages/Python/lldbsuite/test/benchmarks/disassembly/TestDoAttachThenDisassembly.py
+++ b/lldb/packages/Python/lldbsuite/test/benchmarks/disassembly/TestDoAttachThenDisassembly.py
@@ -9,6 +9,7 @@ from __future__ import print_function
 import os, sys
 import lldb
 from lldbsuite.test.lldbbench import *
+from lldbsuite.test import configuration
 
 class AttachThenDisassemblyBench(BenchBase):
 
@@ -16,11 +17,11 @@ class AttachThenDisassemblyBench(BenchBase):
 
     def setUp(self):
         BenchBase.setUp(self)
-        if lldb.bmExecutable:
-            self.exe = lldb.bmExecutable
+        if configuration.bmExecutable:
+            self.exe = configuration.bmExecutable
         else:
             self.exe = lldbtest_config.lldbExec
-        self.count = lldb.bmIterationCount
+        self.count = configuration.bmIterationCount
         if self.count <= 0:
             self.count = 10