| Index: client/common_lib/base_barrier_unittest.py
 | 
| diff --git a/client/common_lib/base_barrier_unittest.py b/client/common_lib/base_barrier_unittest.py
 | 
| index 52d8e1734dae08eb36684aa9ba6bc48ad101ac75..71ea53842f958c489c9cff59852b2b186a73f6cc 100755
 | 
| --- a/client/common_lib/base_barrier_unittest.py
 | 
| +++ b/client/common_lib/base_barrier_unittest.py
 | 
| @@ -5,7 +5,7 @@ __author__ = """Ashwin Ganti (aganti@google.com)"""
 | 
|  import os, sys, socket, errno, unittest, threading
 | 
|  from time import time, sleep
 | 
|  import common
 | 
| -from autotest_lib.client.common_lib import error, barrier
 | 
| +from autotest_lib.client.common_lib import error, barrier, base_barrier
 | 
|  from autotest_lib.client.common_lib.test_utils import mock
 | 
|  
 | 
|  
 | 
| @@ -46,15 +46,14 @@ class barrier_test(unittest.TestCase):
 | 
|  
 | 
|  
 | 
|      def test_get_host_from_id(self):
 | 
| -        b = barrier.barrier('127.0.0.1#', 'testgethost', 100)
 | 
| -
 | 
| -        hostname = b._get_host_from_id('my_host')
 | 
| +        hostname = base_barrier.get_host_from_id('my_host')
 | 
|          self.assertEqual(hostname, 'my_host')
 | 
|  
 | 
| -        hostname = b._get_host_from_id('my_host#')
 | 
| +        hostname = base_barrier.get_host_from_id('my_host#')
 | 
|          self.assertEqual(hostname, 'my_host')
 | 
|  
 | 
| -        self.assertRaises(error.BarrierError, b._get_host_from_id, '#my_host')
 | 
| +        self.assertRaises(error.BarrierError,
 | 
| +                          base_barrier.get_host_from_id, '#my_host')
 | 
|  
 | 
|  
 | 
|      def test_update_timeout(self):
 | 
| 
 |