diff --git a/edx_ace/tests/channel/test_channel_helpers.py b/edx_ace/tests/channel/test_channel_helpers.py
index ef0c2ca..0fa643b 100644
--- a/edx_ace/tests/channel/test_channel_helpers.py
+++ b/edx_ace/tests/channel/test_channel_helpers.py
@@ -8,6 +8,7 @@
 from edx_ace.channel import ChannelMap, ChannelType, get_channel_for_message
 from edx_ace.channel.braze import BrazeEmailChannel
 from edx_ace.channel.file import FileEmailChannel
+from edx_ace.channel.push_notification import PushNotificationChannel
 from edx_ace.errors import UnsupportedChannelError
 from edx_ace.message import Message
 from edx_ace.recipient import Recipient
@@ -45,6 +46,7 @@ def test_get_channel_for_message(self):
         channel_map = ChannelMap([
             ['file_email', FileEmailChannel()],
             ['braze_email', BrazeEmailChannel()],
+            ['push_notifications', PushNotificationChannel()],
         ])
 
         transactional_msg = Message(options={'transactional': True}, **self.msg_kwargs)
@@ -58,6 +60,7 @@ def test_get_channel_for_message(self):
             assert isinstance(get_channel_for_message(ChannelType.EMAIL, transactional_msg), FileEmailChannel)
             assert isinstance(get_channel_for_message(ChannelType.EMAIL, transactional_campaign_msg), BrazeEmailChannel)
             assert isinstance(get_channel_for_message(ChannelType.EMAIL, info_msg), BrazeEmailChannel)
+            assert isinstance(get_channel_for_message(ChannelType.PUSH, info_msg), PushNotificationChannel)
 
             with self.assertRaises(UnsupportedChannelError):
                 assert get_channel_for_message(ChannelType.PUSH, transactional_msg)