diff --git a/tests/func/test_install.py b/tests/func/test_install.py
index 1ff4ca23a..b4e07f4b8 100644
--- a/tests/func/test_install.py
+++ b/tests/func/test_install.py
@@ -39,7 +39,7 @@ class TestInstall(object):
         expected_script = (
             "#!/bin/sh\n"
             "echo hello\n"
-            '[ "$3" == "0" ]'
+            '[ "$3" = "0" ]'
             ' || [ -z "$(git ls-files .dvc)" ]'
             " || exec dvc checkout\n"
         )
@@ -53,7 +53,7 @@ class TestInstall(object):
 
         expected_script = (
             "#!/bin/sh\n"
-            '[ "$3" == "0" ]'
+            '[ "$3" = "0" ]'
             ' || [ -z "$(git ls-files .dvc)" ]'
             " || exec dvc checkout\n"
         )