diff --git a/tests/test_run.py b/tests/test_run.py
index 282455116..b0e26a92a 100644
--- a/tests/test_run.py
+++ b/tests/test_run.py
@@ -187,7 +187,10 @@ class TestRunDeterministic(TestDvc):
     def test(self):
 
         def _run(target, deterministic=False):
-            from shlex import shlex
+            try:  # py3
+                from shlex import quote
+            except ImportError:  # py2
+                from pipes import quote
 
             command = [
                 'dvc', 'run', '--verbose', '-f', '{}.dvc'.format(target), '-d', 'file1', '-o', target,