diff --git a/dig.lua b/dig.lua
index 4b55018..fe585ed 100644
--- a/dig.lua
+++ b/dig.lua
@@ -18,6 +18,12 @@ rednet.send(receiverID, data)
 while true do
     id, msg = rednet.receive()
     if id == receiverID then
-        io.write(msg .. "\n")
+        if msg == nil then
+            print("Hostname not found!")
+            break
+        else
+            io.write(msg .. "\n")
+            break
+        end
     end
 end
\ No newline at end of file
diff --git a/lns_server.lua b/lns_server.lua
index 7fd1ac6..91b4b9e 100644
--- a/lns_server.lua
+++ b/lns_server.lua
@@ -48,7 +48,7 @@ while true do
 
         -- check if hostname in data
         if request.hostname == nil then
-            rednet.send(client_id, "not found")
+            rednet.send(client_id, nil)
         else
             rednet.send(client_id, data[request.hostname])
         end