diff --git a/src/angular/modal.ts b/src/angular/modal.ts
index 0f48eea..26d4207 100644
--- a/src/angular/modal.ts
+++ b/src/angular/modal.ts
@@ -1,7 +1,7 @@
 export class ModalMock {
     public static instance(): any {
         let _dismissCallback: Function;
-        let instance = jasmine.createSpyObj('Modal', ['present', 'dismiss', 'onDidDismiss']);
+        let instance = jasmine.createSpyObj('Modal', ['present', 'dismiss', 'onDidDismiss', 'onWillDismiss']);
         instance.present.and.returnValue(Promise.resolve());
 
         instance.dismiss.and.callFake(() => {
@@ -12,6 +12,10 @@ export class ModalMock {
         instance.onDidDismiss.and.callFake((callback: Function) => {
             _dismissCallback = callback;
         });
+        
+        instance.onWillDismiss.and.callFake((callback: Function) => {
+            _dismissCallback = callback;
+        });
 
         return instance;
     }