diff --git a/test/directives/ng_controller_spec.dart b/test/directives/ng_controller_spec.dart
index 5e59f64be..2664ac05c 100644
--- a/test/directives/ng_controller_spec.dart
+++ b/test/directives/ng_controller_spec.dart
@@ -12,7 +12,10 @@ class MainController {
 
 
 main() {
-  beforeEach(module(angularModule));
+  beforeEach(module((AngularModule module) {
+    module.type(MainController, MainController);
+    angularModule(module);
+  }));
 
   describe('NgController', () {
     var compile, element, rootScope;