diff --git a/main.ts b/main.ts
index 372ac8f..6379041 100755
--- a/main.ts
+++ b/main.ts
@@ -2,6 +2,7 @@
 
 import fs from 'fs-extra';
 import mustache from 'mustache';
+import path from 'path';
 import { program } from 'commander';
 
 import { log, logSuccess, logInfo, logWarning, logError, colors } from './src/utils/logger';
@@ -26,7 +27,11 @@ program
   .version('1.0.0')
   .option('-i, --input <file>', 'Archivo OpenAPI/Swagger (yaml o json)', 'swagger.yaml')
   .option('-o, --output <dir>', 'Directorio de salida', './src/app')
-  .option('-t, --templates <dir>', 'Directorio de templates personalizados', './templates')
+  .option(
+    '-t, --templates <dir>',
+    'Directorio de templates personalizados',
+    path.join(__dirname, 'templates')
+  )
   .option('--skip-install', 'No instalar dependencias')
   .option('--dry-run', 'Simular sin generar archivos')
   .parse(process.argv);
diff --git a/package.json b/package.json
index 4d63162..0c43043 100644
--- a/package.json
+++ b/package.json
@@ -2,14 +2,14 @@
   "name": "openapi-clean-arch-generator",
   "version": "1.0.0",
   "description": "Generador de código Angular con Clean Architecture desde OpenAPI/Swagger",
-  "main": "dist/generate.js",
+  "main": "dist/main.js",
   "bin": {
-    "generate-clean-arch": "./dist/generate.js"
+    "generate-clean-arch": "./dist/main.js"
   },
   "scripts": {
-    "build": "tsc",
+    "build": "tsc && cp -r templates dist/",
     "prepublishOnly": "npm run build",
-    "generate": "node dist/generate.js",
+    "generate": "node dist/main.js",
     "generate:dev": "ts-node main.ts",
     "lint": "eslint 'main.ts' 'src/**/*.ts' -f unix",
     "lint:fix": "eslint 'main.ts' 'src/**/*.ts' --fix -f unix",