diff --git a/src/Form/WorkflowSmsNotificationForm.php b/src/Form/WorkflowSmsNotificationForm.php
index 9080c17..338097b 100644
--- a/src/Form/WorkflowSmsNotificationForm.php
+++ b/src/Form/WorkflowSmsNotificationForm.php
@@ -72,6 +72,26 @@ class WorkflowSmsNotificationForm extends EntityForm {
       '#default_value' => $default_message['value'],
     ];
 
+    // Token support.
+    if (\Drupal::moduleHandler()->moduleExists('token')) {
+      $form['tokens'] = [
+        '#title' => $this->t('Tokens'),
+        '#type' => 'container',
+        '#states' => [
+          'invisible' => [
+            'input[name="use_token"]' => ['checked' => FALSE],
+          ],
+        ],
+      ];
+      $form['tokens']['help'] = [
+        '#theme' => 'token_tree_link',
+        '#token_types' => ['node', 'workflow_transition', 'workflow_scheduled_transition', 'term', 'paragraph', 'comment'],
+        // '#token_types' => 'all'
+        '#global_types' => FALSE,
+        '#dialog' => TRUE,
+      ];
+    }
+    
     $form['recipients'] = [
       '#type' => 'details',
       '#title' => $this->t('Recipients'),
diff --git a/workflow_sms_notification.module b/workflow_sms_notification.module
index e3da280..84751ab 100644
--- a/workflow_sms_notification.module
+++ b/workflow_sms_notification.module
@@ -30,7 +30,7 @@ function workflow_sms_notification_workflow($op, WorkflowTransitionInterface $tr
         $token = Drupal::token();
         // Setup the sms message content.
         $message = $notification->getMessage();
-        $message['value'] = $token->replace($message['value'], [$entity->getEntityTypeId() => $entity]);
+         $message['value'] = $token->replace($message['value'], [$entity->getEntityTypeId() => $entity, 'workflow_transition' => $transition, 'workflow_scheduled_transition' => $transition]);
         $message = (string) check_markup($message['value'], $message['format']);
 
         // Figure out who the email should be going to.