1 files changed, 8 insertions, 17 deletions
diff --git a/test/ngScenario/dslSpec.js b/test/ngScenario/dslSpec.js
index fee5c3b5..eef8c907 100644
--- a/test/ngScenario/dslSpec.js
+++ b/test/ngScenario/dslSpec.js
@@ -168,23 +168,14 @@ describe("angular.scenario.dsl", function() {
     });
 
     describe('location', function() {
-      beforeEach(function() {
-        $window.angular.injector = function() {
-          return {
-            get: function(serviceId) {
-              if (serviceId == '$location') {
-                return {
-                  url: function() {return '/path?search=a#hhh';},
-                  path: function() {return '/path';},
-                  search: function() {return {search: 'a'};},
-                  hash: function() {return 'hhh';}
-                };
-              }
-              throw new Error('unknown service id ' + serviceId);
-            }
-          };
-        };
-      });
+      beforeEach(inject(function($injector) {
+        angular.extend($injector.get('$location'), {
+          url: function() {return '/path?search=a#hhh';},
+          path: function() {return '/path';},
+          search: function() {return {search: 'a'};},
+          hash: function() {return 'hhh';}
+        });
+      }));
 
       it('should return full url', function() {
         $root.dsl.browser().location().url();