diff --git a/web/pgadmin/tools/sqleditor/tests/test_poll_query_tool.py b/web/pgadmin/tools/sqleditor/tests/test_poll_query_tool.py
index 275ed9c..d007e86 100644
--- a/web/pgadmin/tools/sqleditor/tests/test_poll_query_tool.py
+++ b/web/pgadmin/tools/sqleditor/tests/test_poll_query_tool.py
@@ -100,7 +100,7 @@ NOTICE:  Hello, world!
             response_data = json.loads(response.data.decode('utf-8'))
 
             # Check the returned messages
-            self.assertEquals(self.expected_message[cnt],
+            self.assertIn(self.expected_message[cnt],
                               response_data['data']['additional_messages'])
             # Check the output
             self.assertEquals(self.expected_result[cnt],