fix tests for lit correlation

1 files changed, 1 insertions, 0 deletions

diff --git a/tests/unit/computations/test_correlation.py b/tests/unit/computations/test_correlation.py
index 8f3ef25..26301eb 100644
--- a/tests/unit/computations/test_correlation.py
+++ b/tests/unit/computations/test_correlation.py
@@ -302,6 +302,7 @@ class TestCorrelation(TestCase):
         db value if corr coefficient is empty
         """
         database_instance = mock.Mock()
+        database_instance.cursor.return_value = DataBase()
         database_instance.execute.return_value.fetchone.return_value = None
 
         lit_results = fetch_lit_correlation_data(conn=database_instance,