diff --git a/calls.php b/calls.php
index e3f0dbf..2a8d47b 100644
--- a/calls.php
+++ b/calls.php
@@ -36,7 +36,7 @@ if ($_SERVER['REQUEST_METHOD'] === 'GET') {
             } elseif ($_GET['value'] == 'domain') {
                 $idle->viewMoreDomainModal($_GET['id']);//View more details modal
             }
-        } elseif ($_GET['type'] == 'dns_search') {
+        } elseif ($_GET['type'] == 'A' || $_GET['type'] == 'AAAA') {
             $idle->getIpForDomain($_GET['hostname'], $_GET['type']);
         } elseif ($_GET['type'] == 'check_up') {
             echo $idle->checkIsUp($_GET['host']);