| Index: client/tests/kvm/tests/steps.py
 | 
| diff --git a/client/tests/kvm/tests/steps.py b/client/tests/kvm/tests/steps.py
 | 
| index 5d4ed25068fd7face3b44e8ced680239ce69582c..6f782f562d09f7b1581217c7e148bea732931996 100644
 | 
| --- a/client/tests/kvm/tests/steps.py
 | 
| +++ b/client/tests/kvm/tests/steps.py
 | 
| @@ -181,7 +181,7 @@ def barrier_2(vm, words, params, debug_dir, data_scrdump_filename,
 | 
|  
 | 
|  
 | 
|  def run_steps(test, params, env):
 | 
| -    vm = env.get_vm(params.get("main_vm"))
 | 
| +    vm = kvm_utils.env_get_vm(env, params.get("main_vm"))
 | 
|      if not vm:
 | 
|          raise error.TestError("VM object not found in environment")
 | 
|      if not vm.is_alive():
 | 
| 
 |