diff --git a/src/pquant/pruning_methods/fitcompress.py b/src/pquant/pruning_methods/fitcompress.py
index f2b3b5c..9e5b314 100644
--- a/src/pquant/pruning_methods/fitcompress.py
+++ b/src/pquant/pruning_methods/fitcompress.py
@@ -26,7 +26,7 @@ def get_hard_mask(self, weight=None):
     def pre_epoch_function(self, epoch, total_epochs):
         pass
 
-    def calculate_additional_loss(*args, **kwargs):
+    def calculate_additional_loss(self):
         return 0
 
     def pre_finetune_function(self):