diff --git a/docs/dgeni-package/index.js b/docs/dgeni-package/index.js
index 15fe3ae7cf..e46daffbc5 100644
--- a/docs/dgeni-package/index.js
+++ b/docs/dgeni-package/index.js
@@ -62,7 +62,7 @@ module.exports = new Package('angular', [jsdocPackage, nunjucksPackage, linksPac
   ];
 
   readTypeScriptModules.sourceFiles = [
-    '*/*.@(js|es6|ts])',
+    '*/*.@(js|es6|ts)',
     '*/src/**/*.@(js|es6|ts)'
   ];
   readTypeScriptModules.basePath = 'modules';
diff --git a/docs/dgeni-package/processors/createTypeDefinitionFile.js b/docs/dgeni-package/processors/createTypeDefinitionFile.js
index 95b98f1df4..066ebd9110 100644
--- a/docs/dgeni-package/processors/createTypeDefinitionFile.js
+++ b/docs/dgeni-package/processors/createTypeDefinitionFile.js
@@ -14,11 +14,13 @@ module.exports = function createTypeDefinitionFile() {
         modules: []
       };
       _.forEach(docs, function(doc) {
-        if ( doc.docType === 'module' ) {
+        // The shape of the public API is determined by what is reexported into
+        // angular2/angular2.
+        if (doc.id === 'angular2/angular2') {
           typeDefDoc.modules.push(doc);
         }
       });
       docs.push(typeDefDoc);
     }
   };
-};
\ No newline at end of file
+};