diff --git a/pyqtorch/noise/readout.py b/pyqtorch/noise/readout.py
index ecade935..f7606a76 100644
--- a/pyqtorch/noise/readout.py
+++ b/pyqtorch/noise/readout.py
@@ -159,7 +159,7 @@ def create_confusion_matrices(noise_matrix: Tensor, error_probability: float) ->
 
 class ReadoutInterface(ABC):
     @singledispatchmethod
-    def apply(self, input_to_corrupt, n_shots: int):
+    def apply(self, input_to_corrupt, n_shots):
         raise NotImplementedError