2 files changed, 11 insertions, 0 deletions
diff --git a/ipatests/test_ipapython/test_dn.py b/ipatests/test_ipapython/test_dn.py
index 69ccbde8e..14aaeafb5 100644
--- a/ipatests/test_ipapython/test_dn.py
+++ b/ipatests/test_ipapython/test_dn.py
@@ -2,8 +2,13 @@
 
 import unittest
 
+import six
+
 from ipapython.dn import *
 
+if six.PY3:
+    unicode = str
+
 def expected_class(klass, component):
     if klass is AVA:
         if component == 'self':
diff --git a/ipatests/test_ipapython/test_ssh.py b/ipatests/test_ipapython/test_ssh.py
index 913ff8e6b..9826a34c3 100644
--- a/ipatests/test_ipapython/test_ssh.py
+++ b/ipatests/test_ipapython/test_ssh.py
@@ -21,10 +21,16 @@ Test the `ipapython/ssh.py` module.
 """
 
 import base64
+
+import six
 import nose
 
 from ipapython import ssh
 
+if six.PY3:
+    unicode = str
+
+
 def make_public_key_checker(pk, out):
     def check_public_key():
         try: