diff --git a/main.js b/main.js
index 3ae105c..8ff524a 100644
--- a/main.js
+++ b/main.js
@@ -20,6 +20,7 @@ async function main() {
 
         if (directory) {
             process.chdir(directory)
+            core.saveState("directory", directory)
         }
 
         if (key) {