diff --git a/tests/integration/api_build_test.py b/tests/integration/api_build_test.py
index 62e93a7384..0f560159b3 100644
--- a/tests/integration/api_build_test.py
+++ b/tests/integration/api_build_test.py
@@ -275,7 +275,7 @@ def test_build_container_with_target(self):
             pass
 
         info = self.client.inspect_image('build1')
-        assert not info['Config']['OnBuild']
+        assert 'OnBuild' not in info['Config'] or not info['Config']['OnBuild']
 
     @requires_api_version('1.25')
     def test_build_with_network_mode(self):
diff --git a/tests/ssh/api_build_test.py b/tests/ssh/api_build_test.py
index 20476fc74d..f17c75630f 100644
--- a/tests/ssh/api_build_test.py
+++ b/tests/ssh/api_build_test.py
@@ -266,7 +266,7 @@ def test_build_container_with_target(self):
             pass
 
         info = self.client.inspect_image('build1')
-        assert not info['Config']['OnBuild']
+        assert 'OnBuild' not in info['Config'] or not info['Config']['OnBuild']
 
     @requires_api_version('1.25')
     def test_build_with_network_mode(self):