1 files changed, 4 insertions, 5 deletions
diff --git a/crosperf/benchmark_run_unittest.py b/crosperf/benchmark_run_unittest.py
index 5696c107..4cadc35c 100755
--- a/crosperf/benchmark_run_unittest.py
+++ b/crosperf/benchmark_run_unittest.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python2
+#!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
 # Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
@@ -9,10 +9,9 @@
 
 from __future__ import print_function
 
-import unittest
 import inspect
-
-import mock
+import unittest
+import unittest.mock as mock
 
 import benchmark_run
 
@@ -121,7 +120,7 @@ class BenchmarkRunTest(unittest.TestCase):
         'machine_manager', 'logger_to_use', 'log_level', 'share_cache',
         'dut_config'
     ]
-    arg_spec = inspect.getargspec(benchmark_run.BenchmarkRun.__init__)
+    arg_spec = inspect.getfullargspec(benchmark_run.BenchmarkRun.__init__)
     self.assertEqual(len(arg_spec.args), len(args_list))
     self.assertEqual(arg_spec.args, args_list)