diff --git a/action.js b/action.js
index 7b565fa..d42ca18 100644
--- a/action.js
+++ b/action.js
@@ -22,7 +22,7 @@ writeFileSync(`${libName}/__init__.mpy`, "");
 packageJsonUrls.push([`${libName}/__init__.mpy`, `__init__.mpy`]);
 
 // Generate module folders/files
-for (let modPy of readdirSync(".").filter((f) => f.slice(-2) == "py")) {
+for (let modPy of readdirSync(".").filter((f) => f.slice(-3) == ".py")) {
   const modName = modPy.slice(0, -3);
   const modInitPath = `${modName}/__init__.mpy`;
   const modMpyPath = `${modName}/${modName}.mpy`;