diff --git a/plugins/example.py b/plugins/example.py
index f9d7c31..a0d2d9d 100644
--- a/plugins/example.py
+++ b/plugins/example.py
@@ -160,11 +160,6 @@ def dns(params,authentication):
 def niami(params, authentication):
     domain = params["domain"]
     response = requests.get(f"https://api.handshake.niami.io/domain/{domain}")
-    print(response.text)
-    if response.status_code != 200:
-        return {"rating":"Error fetching rating from Niami.io"}
-    if response.json()["success"] == False:
-        return {"rating":"Error fetching rating from Niami.io"}
     data = response.json()["data"]
     rating = str(data["rating"]["score"]) + " (" + data["rating"]["rarity"] + ")"
     return {"rating":rating}
diff --git a/render.py b/render.py
index 98a3ecf..6987dba 100644
--- a/render.py
+++ b/render.py
@@ -272,6 +272,11 @@ def plugin_output(outputs, returns):
         html += f'<h4 class="card-title">{returns[returnOutput]["name"]}</h4>'
         
         # Get the output
+        if returnOutput not in outputs:
+            html += f'<p>No output</p>'
+            html += f'</div>'
+            html += f'</div>'
+            continue
         output = outputs[returnOutput]
         if returns[returnOutput]["type"] == "list":
             html += f'<ul>'