diff --git a/DTLSSocket/__init__.py b/DTLSSocket/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..cdbe2170d4b23eb2e3ce2162601cfff809a36306
--- /dev/null
+++ b/DTLSSocket/__init__.py
@@ -0,0 +1 @@
+from .DTLSSocket import DTLSSocket as DTLSSocket
diff --git a/tests/testSocket.py b/tests/testSocket.py
index f4ed18f494bbd57344c2f06f3028d14129505e7b..90d716f16be15c6054437d40e91d191b9e6f52a0 100644
--- a/tests/testSocket.py
+++ b/tests/testSocket.py
@@ -1,4 +1,5 @@
-import dtls, time, socket
+from DTLSSocket import dtls
+import time, socket
 
 sock = socket.socket(family=socket.AF_INET6, type=socket.SOCK_DGRAM)
 sock.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_RECVPKTINFO, 1)
diff --git a/tests/testSocketServer.py b/tests/testSocketServer.py
index cf57bbc98f25ed447b5bd00ac08528c59ac3cb03..5ea2c941cbb4adf5ab042abf3a19cfa5464e0f9d 100644
--- a/tests/testSocketServer.py
+++ b/tests/testSocketServer.py
@@ -1,4 +1,5 @@
-import dtls, time, socket
+from DTLSSocket import dtls
+import time, socket
 
 sock = socket.socket(family=socket.AF_INET6, type=socket.SOCK_DGRAM)
 sock.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_RECVPKTINFO, 1)