diff --git a/docker/ops.go b/docker/ops.go
index 9ae49d5..7fdb01d 100644
--- a/docker/ops.go
+++ b/docker/ops.go
@@ -69,6 +69,7 @@ func generate_sbom(message *babashka.Message, image string, username string, pas
 				break
 			}
 		}
+		babashka.WriteInvokeResponse(message, "done");
 		return nil
 	}()
 
@@ -192,7 +193,7 @@ func ProcessMessage(message *babashka.Message) (any, error) {
 					babashka.WriteErrorResponse(message, err)
 				}
 			}
-			return "done", nil
+			return "running", nil
 
 		case "docker.tools/generate-hashes":
 			args := []string{}
diff --git a/main.go b/main.go
index 9f4da6b..094eb9f 100644
--- a/main.go
+++ b/main.go
@@ -28,7 +28,9 @@ func main() {
 			babashka.WriteDescribeResponse(describeres)
 			continue
 		}
-		// TODO don't write done responses when callback is running
-		babashka.WriteInvokeResponse(message, res)
+
+		if res != "running" {
+			babashka.WriteInvokeResponse(message, res)
+		}
 	}
 }