diff --git a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_message/test_whatsapp_message.py b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_message/test_whatsapp_message.py
index cb09e84..1ccfd9a 100644
--- a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_message/test_whatsapp_message.py
+++ b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_message/test_whatsapp_message.py
@@ -3,10 +3,10 @@
 # See license.txt
 
 # import frappe
-from frappe.tests.utils import FrappeTestCase
+from frappe.tests import UnitTestCase
 
 
-class TestWhatsAppMessage(FrappeTestCase):
+class TestWhatsAppMessage(UnitTestCase):
     """Test whatsapp messages."""
 
     pass
diff --git a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification/test_whatsapp_notification.py b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification/test_whatsapp_notification.py
index 0246efe..3dcfd68 100644
--- a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification/test_whatsapp_notification.py
+++ b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification/test_whatsapp_notification.py
@@ -2,8 +2,8 @@
 # See license.txt
 
 # import frappe
-from frappe.tests.utils import FrappeTestCase
+from frappe.tests import UnitTestCase
 
 
-class TestWhatsAppNotification(FrappeTestCase):
+class TestWhatsAppNotification(UnitTestCase):
 	pass
diff --git a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification_log/test_whatsapp_notification_log.py b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification_log/test_whatsapp_notification_log.py
index 05d641f..b7b8226 100644
--- a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification_log/test_whatsapp_notification_log.py
+++ b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_notification_log/test_whatsapp_notification_log.py
@@ -2,8 +2,8 @@
 # See license.txt
 
 # import frappe
-from frappe.tests.utils import FrappeTestCase
+from frappe.tests import UnitTestCase
 
 
-class TestWhatsAppNotificationLog(FrappeTestCase):
+class TestWhatsAppNotificationLog(UnitTestCase):
 	pass
diff --git a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_settings/test_whatsapp_settings.py b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_settings/test_whatsapp_settings.py
index 13eff81..cd69fe7 100644
--- a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_settings/test_whatsapp_settings.py
+++ b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_settings/test_whatsapp_settings.py
@@ -2,8 +2,8 @@
 # See license.txt
 
 # import frappe
-from frappe.tests.utils import FrappeTestCase
+from frappe.tests import UnitTestCase
 
 
-class TestWhatsAppSettings(FrappeTestCase):
+class TestWhatsAppSettings(UnitTestCase):
 	pass
diff --git a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_templates/test_whatsapp_templates.py b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_templates/test_whatsapp_templates.py
index 6650b9c..edcbb7b 100644
--- a/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_templates/test_whatsapp_templates.py
+++ b/frappe_whatsapp/frappe_whatsapp/doctype/whatsapp_templates/test_whatsapp_templates.py
@@ -2,8 +2,8 @@
 # See license.txt
 
 # import frappe
-from frappe.tests.utils import FrappeTestCase
+from frappe.tests import UnitTestCase
 
 
-class TestWhatsAppTemplates(FrappeTestCase):
+class TestWhatsAppTemplates(UnitTestCase):
 	pass