diff --git a/elm/wizard.py b/elm/wizard.py
index d312bad4..ba26ffa0 100644
--- a/elm/wizard.py
+++ b/elm/wizard.py
@@ -61,8 +61,6 @@ def query_vector_db(self, query, limit=100):
             ranked strings/scores outputs.
         """
 
-    def test(self, question):
-        return question
     def engineer_query(self, query, token_budget=None, new_info_threshold=0.7,
                        convo=False):
         """Engineer a query for GPT using the corpus of information
diff --git a/tests/test_postgres.py b/tests/test_postgres.py
index 4063b00f..08e6fef3 100644
--- a/tests/test_postgres.py
+++ b/tests/test_postgres.py
@@ -35,7 +35,8 @@ def query_call(*args, **kwargs):  # pylint: disable=unused-argument
 
 def test_ref_list(mocker):
     """Test to ensure correct response vector db."""
-    wizard_mock = mocker.patch('elm.wizard.EnergyWizardPostgres', autospec=True)
+    wizard_mock = mocker.patch('elm.wizard.EnergyWizardPostgres',
+                               autospec=True)
 
     wizard = wizard_mock.return_value
     wizard.messages = []
@@ -43,7 +44,7 @@ def test_ref_list(mocker):
     wizard.token_budget = 500
     wizard.model = "dummy-model-name"
     wizard.count_tokens = mocker.Mock(return_value=50)
-    wizard.make_ref_list.side_effect= MockClass.ref_call
+    wizard.make_ref_list.side_effect = MockClass.ref_call
     wizard.query_vector_db.side_effect = MockClass.query_call
 
     question = "What is a dummy question?"