diff --git a/Gruntfile.js b/Gruntfile.js
index 39a1edc..fbdedd4 100644
--- a/Gruntfile.js
+++ b/Gruntfile.js
@@ -7,7 +7,7 @@ module.exports = function (grunt) {
   grunt.initConfig({
     ngdocs: {
       options: {
-        scripts: ['angular.js', '../src.js'],
+        scripts: ['angular.js', 'src/directive.js'],
         html5Mode: false
       },
       all: ['src/*.js']
diff --git a/package.json b/package.json
index 748a03f..d0cbb23 100644
--- a/package.json
+++ b/package.json
@@ -4,7 +4,7 @@
   "description": "Basic example for grunt-ngdocs",
   "dependencies": {
     "grunt": "~0.4.1",
-    "grunt-ngdocs": "~0.1.1",
+    "grunt-ngdocs": "~0.2.9",
     "grunt-contrib-connect": "~0.3.0",
     "grunt-contrib-clean": "0.4.1"
   }
diff --git a/src/controller.js b/src/controller.js
new file mode 100644
index 0000000..c55ca76
--- /dev/null
+++ b/src/controller.js
@@ -0,0 +1,10 @@
+'use strict';
+
+/**
+ * @ngdoc controller
+ * @name rfx.controller:RestController
+ * @description
+ * This is a controller which uses the {@link rfx.rest rest} service.
+ */
+angular.module('rfx')
+  .controller('RestController', ['rest', function (rest) {}]);