diff --git a/test/modules/BugReporting.spec.ts b/test/modules/BugReporting.spec.ts
index 8fa36b003..5092445e5 100644
--- a/test/modules/BugReporting.spec.ts
+++ b/test/modules/BugReporting.spec.ts
@@ -135,18 +135,18 @@ describe('Testing BugReporting Module', () => {
 
   it('should call the native method setOnDismissHandler with a function', () => {
     const callback = jest.fn();
-    BugReporting.onSDKDismissedHandler(callback);
+    BugReporting.onDismissHandler(callback);
 
     expect(NativeBugReporting.setOnDismissHandler).toBeCalledTimes(1);
     expect(NativeBugReporting.setOnDismissHandler).toBeCalledWith(callback);
   });
 
   it('should invoke callback on emitting the event IBGpostInvocationHandler', () => {
-    const dismissType = 'cancel';
-    const reportType = 'bug';
+    const dismissType = Instabug.dismissType.cancel;
+    const reportType = BugReporting.reportType.bug;
     const callback = jest.fn();
 
-    BugReporting.onSDKDismissedHandler(callback);
+    BugReporting.onDismissHandler(callback);
     IBGEventEmitter.emit(IBGConstants.ON_SDK_DISMISSED_HANDLER, {
       dismissType: dismissType,
       reportType: reportType,