From 0742210fc7eaa54d8068bfd18414158dc9b33672 Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Tue, 18 Aug 2026 16:35:11 -0700
Subject: Ruff cleanup

---
 tests/models/network/test_rdns.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'tests/models/network/test_rdns.py')

diff --git a/tests/models/network/test_rdns.py b/tests/models/network/test_rdns.py
index e56c494..5c3b024 100644
--- a/tests/models/network/test_rdns.py
+++ b/tests/models/network/test_rdns.py
@@ -1,12 +1,13 @@
-from generalresearch.models.network.rdns.execute import execute_rdns
 import faker
 
-from generalresearch.models.network.tool_run import ToolName, ToolClass
+from generalresearch.managers.network.tool_run import ToolRunManager
+from generalresearch.models.network.rdns.execute import execute_rdns
+from generalresearch.models.network.tool_run import ToolClass, ToolName
 
 fake = faker.Faker()
 
 
-def test_execute_rdns_grl(toolrun_manager):
+def test_execute_rdns_grl(toolrun_manager: ToolRunManager):
     ip = "65.19.129.53"
     run = execute_rdns(ip=ip)
     assert run.tool_name == ToolName.DIG
@@ -20,7 +21,7 @@ def test_execute_rdns_grl(toolrun_manager):
     toolrun_manager.create_rdns_run(run)
 
 
-def test_execute_rdns_none(toolrun_manager):
+def test_execute_rdns_none(toolrun_manager: ToolRunManager):
     ip = fake.ipv6()
     run = execute_rdns(ip)
     result = run.parsed
-- 
cgit v1.2.3