diff --git a/main.js b/main.js
index ccc4b8d..3ac9d2e 100644
--- a/main.js
+++ b/main.js
@@ -78,6 +78,7 @@ async function main() {
             image,
             "sleep", "inf"
         ])
+        core.saveState("container", container)
         core.endGroup()
 
         core.startGroup("Start container")