diff --git a/commands/setup_theme.js b/commands/setup_theme.js
index b2d6529..c219403 100644
--- a/commands/setup_theme.js
+++ b/commands/setup_theme.js
@@ -47,9 +47,9 @@ async function setup_theme(themeName) {
       return false;
     }
 
-    // Ensure the current working directory ends with wp-themes
+    // Ensure the current working directory ends themes
     const cwd = process.cwd();
-    if (!cwd.endsWith('wp-themes')) {
+    if (!cwd.endsWith('wp-content/themes')) {
       console.log(
         chalk.red('\nThis command needs to be run inside a wp-themes folder.')
       );
diff --git a/package.json b/package.json
index 6d2a261..552e846 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@webredone/trb-cli",
-  "version": "3.0.3",
+  "version": "3.0.4",
   "description": "TRB-CLI is a handy Node.js CLI that automates the process of managing blocks in a Theme Redone WordPress theme",
   "main": "index.js",
   "type": "module",