1 files changed, 4 insertions, 2 deletions

diff --git a/test/stufftest.lua b/test/stufftest.lua
index 5eb8005..6bd1f91 100644
--- a/test/stufftest.lua
+++ b/test/stufftest.lua
@@ -1,11 +1,13 @@
+smtp = require("smtp")
+
 function test_dot(original, right)
-    local result, n = socket.smtp.dot(2, original)
+    local result, n = smtp.dot(2, original)
     assert(result == right, "->" .. result .. "<-")
     print("ok")
 end
 
 function test_stuff(original, right)
-    local result, n = socket.smtp.dot(2, original)
+    local result, n = smtp.dot(2, original)
     assert(result == right, "->" .. result .. "<-")
     print("ok")
 end