1 files changed, 2 insertions, 6 deletions
diff --git a/testsuite/common.py b/testsuite/common.py
index fc79e73e7..3aeeaf5f2 100644
--- a/testsuite/common.py
+++ b/testsuite/common.py
@@ -3,12 +3,7 @@ import re
 import sys
 import unittest
 from mock import patch, MagicMock, _patch, DEFAULT
-
-try:
-    from functools import wraps
-except ImportError:
-    def wraps(wrapped):
-        return lambda f: f
+from Bcfg2.Compat import wraps
 
 datastore = "/"
 
@@ -33,6 +28,7 @@ try:
 except ImportError:
     has_django = False
 
+
 try:
     from mock import call
 except ImportError: