fix speculate tests

blacklanternsecurity · Feb 7, 2024 · 5f8d266 · 5f8d266
1 parent e5d88dc
commit 5f8d266
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bbot/test/test_step_2/module_tests/test_module_speculate.py b/bbot/test/test_step_2/module_tests/test_module_speculate.py
@@ -30,8 +30,8 @@ class TestSpeculate_OpenPorts(ModuleTestBase):
     async def setup_before_prep(self, module_test):
         module_test.mock_dns(
             {
-                "evilcorp.com": {"A": "127.0.254.1"},
-                "asdf.evilcorp.com": {"A": "127.0.254.2"},
+                "evilcorp.com": {"A": ["127.0.254.1"]},
+                "asdf.evilcorp.com": {"A": ["127.0.254.2"]},
             }
         )