diff --git a/pyproject.toml b/pyproject.toml
index b8f7bec..243b49f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-workflowcheckpointing"
 description = "Automatically creates checkpoints during workflow execution. If If an workflow is canceled or ComfyUI crashes mid-execution, then these checkpoints are used when the workflow is re-queued to resume execution with minimal progress loss."
-version = "1.1.0"
+version = "1.1.1"
 license = { file = "LICENSE" }
 
 [project.urls]
diff --git a/workflowcheckpointing.py b/workflowcheckpointing.py
index 1f1309a..0530cdf 100644
--- a/workflowcheckpointing.py
+++ b/workflowcheckpointing.py
@@ -325,13 +325,6 @@ async def fetch_remote_files(remote_files, uid=None):
     if len(fetches) > 0:
         await asyncio.gather(*fetches)
 
-completion_futures = {}
-def add_future(json_data):
-    index = max(completion_futures.keys())
-    json_data['extra_data']['completion_future'] = index
-    return json_data
-server.PromptServer.instance.add_on_prompt_handler(add_future)
-
 prompt_route = next(filter(lambda x: x.path  == '/prompt' and x.method == 'POST',
                            server.PromptServer.instance.routes))
 original_post_prompt = prompt_route.handler