diff --git a/patchwork/steps/ModifyCodePB/ModifyCodePB.py b/patchwork/steps/ModifyCodePB/ModifyCodePB.py
index f5d2f171..39bfbd6c 100644
--- a/patchwork/steps/ModifyCodePB/ModifyCodePB.py
+++ b/patchwork/steps/ModifyCodePB/ModifyCodePB.py
@@ -9,8 +9,8 @@ class ModifyCodePB(Step, input_class=ModifyCodePBInputs, output_class=ModifyCode
     def __init__(self, inputs: dict):
         super().__init__(inputs)
         self.file_path = inputs["file_path"]
-        self.start_line = inputs["start_line"]
-        self.end_line = inputs["end_line"]
+        self.start_line = inputs.get("start_line")
+        self.end_line = inputs.get("end_line")
         self.patch = inputs["new_code"]
 
     def run(self) -> dict:
diff --git a/pyproject.toml b/pyproject.toml
index 3b4373d8..ee207aa4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "patchwork-cli"
-version = "0.0.51"
+version = "0.0.52.dev0"
 description = ""
 authors = ["patched.codes"]
 license = "AGPL"