diff --git a/pytest_arraydiff/plugin.py b/pytest_arraydiff/plugin.py
index 8bc4443..591e53b 100755
--- a/pytest_arraydiff/plugin.py
+++ b/pytest_arraydiff/plugin.py
@@ -36,6 +36,7 @@
 import shutil
 import tempfile
 import warnings
+from distutils.version import StrictVersion
 
 import six
 from six.moves.urllib.request import urlopen
@@ -212,7 +213,11 @@ def __init__(self, config, reference_dir=None, generate_dir=None, default_format
 
     def pytest_runtest_setup(self, item):
 
-        compare = item.keywords.get('array_compare')
+        if StrictVersion(pytest.__version__) < StrictVersion("3.6"):
+            compare = item.get_marker('array_compare')
+        else:
+            compare = item.get_closest_marker('array_compare')
+        
 
         if compare is None:
             return