diff --git a/ecg_data_manager/modules/visualization.py b/ecg_data_manager/modules/visualization.py
index fc4d313..94b9f1a 100644
--- a/ecg_data_manager/modules/visualization.py
+++ b/ecg_data_manager/modules/visualization.py
@@ -507,7 +507,7 @@ def hide_widgets(b):  # pylint: disable=unused-argument
 
         return widgets_box
 
-    def save_diagnosis(  # pylint: disable=too-many-locals, too-many-positional-arguments
+    def save_diagnosis(  # pylint: disable=too-many-locals, too-many-arguments, too-many-positional-arguments
         self,
         user_id,
         document_id,