1 files changed, 7 insertions, 3 deletions
diff --git a/src/lib/tlslite/messages.py b/src/lib/tlslite/messages.py
index afccc793a..1eb20fed7 100755
--- a/src/lib/tlslite/messages.py
+++ b/src/lib/tlslite/messages.py
@@ -8,8 +8,12 @@ from constants import *
 from X509 import X509
 from X509CertChain import X509CertChain
 
-import sha
-import md5
+try:
+    import hashlib as sha
+    import hashlib as md5
+except:
+    import sha
+    import md5
 
 class RecordHeader3:
     def __init__(self):
@@ -558,4 +562,4 @@ class ApplicationData(Msg):
         return self
 
     def write(self):
-        return self.bytes
-\ No newline at end of file
+        return self.bytes