diff --git a/.gitignore b/.gitignore
index 72364f9..ac05bfb 100644
--- a/.gitignore
+++ b/.gitignore
@@ -87,3 +87,6 @@ ENV/
 
 # Rope project settings
 .ropeproject
+
+# Vim
+.*.sw[op]
diff --git a/pytest_pudb.py b/pytest_pudb.py
index 1f7a592..7e07dfd 100644
--- a/pytest_pudb.py
+++ b/pytest_pudb.py
@@ -12,13 +12,20 @@ def pytest_addoption(parser):
 
 
 def pytest_configure(config):
+
     pudb_wrapper = PuDBWrapper(config)
 
     if config.getvalue("usepudb"):
         config.pluginmanager.register(pudb_wrapper, 'pudb_wrapper')
 
     pudb_wrapper.mount()
-    config._cleanup.append(pudb_wrapper.unmount)
+
+    # pytest added `add_cleaup` function since v9, so keeping it backward
+    # compatible.
+    if hasattr(config, '_cleanup'):
+        config._cleanup.append(pudb_wrapper.unmount)
+    else:
+        config.add_cleanup(pudb_wrapper.unmount)
 
 
 class PuDBWrapper(object):