diff --git a/index.js b/index.js
index <HASH>..<HASH> 100644
--- a/index.js
+++ b/index.js
@@ -2,12 +2,15 @@
 
 var YuidocCompiler = require('broccoli-yuidoc');
 var mergeTrees = require('broccoli-merge-trees');
+var fs = require('fs');
 
 module.exports = {
   name: 'ember-cli-yuidoc',
 
-  postprocessTree: function(type, workingTree){
-    var yuidocTree = new YuidocCompiler('addon', {
+  postprocessTree: function(type, workingTree) {
+    var codeFolder = fs.existsSync('addon') ? 'addon' : 'app';
+
+    var yuidocTree = new YuidocCompiler(codeFolder, {
       srcDir: '/',
       destDir: '/docs',
       yuidoc: {}
If /addon exists, watch it. If not, watch /app