diff --git a/framework/tooling/mhtml/runner.go b/framework/tooling/mhtml/runner.go
index 5372cce..c28e320 100644
--- a/framework/tooling/mhtml/runner.go
+++ b/framework/tooling/mhtml/runner.go
@@ -55,4 +55,11 @@ func main() {
 		fmt.Printf("Error running task command: %v\n", err)
 		return
 	}
+
+	err = cmd.Wait()
+
+	if err != nil {
+		fmt.Printf("Error waiting for task command: %v\n", err)
+		return
+	}
 }