diff --git a/monkey/infection_monkey/post_breach/pba.py b/monkey/infection_monkey/post_breach/pba.py
index 9222d5e1a..302830e74 100644
--- a/monkey/infection_monkey/post_breach/pba.py
+++ b/monkey/infection_monkey/post_breach/pba.py
@@ -1,6 +1,6 @@
 import logging
 import subprocess
-from typing import Iterable
+from typing import Dict, Iterable
 
 from common.utils.attack_utils import ScanStatus
 from infection_monkey.i_puppet.i_puppet import PostBreachData
@@ -30,7 +30,7 @@ class PBA:
         self.pba_data = []
         self.telemetry_messenger = telemetry_messenger
 
-    def run(self) -> Iterable[PostBreachData]:
+    def run(self, options: Dict) -> Iterable[PostBreachData]:
         """
         Runs post breach action command
         """