diff --git a/packages/Webkul/Admin/src/Resources/views/settings/workflows/create.blade.php b/packages/Webkul/Admin/src/Resources/views/settings/workflows/create.blade.php
index a07d00027..c571f7100 100644
--- a/packages/Webkul/Admin/src/Resources/views/settings/workflows/create.blade.php
+++ b/packages/Webkul/Admin/src/Resources/views/settings/workflows/create.blade.php
@@ -329,11 +329,17 @@ class="control"
                     </div>
 
                     <div class="form-group" v-if="matchedAttribute.type == 'select' || matchedAttribute.type == 'radio' || matchedAttribute.type == 'lookup'">
-                        <select :name="['actions[' + index + '][value]']" class="control" v-model="action.value">
+                        <select :name="['actions[' + index + '][value]']" class="control" v-model="action.value" v-if="! matchedAttribute.lookup_type">
                             <option v-for='option in matchedAttribute.options' :value="option.id">
                                 @{{ option.name }}
                             </option>
                         </select>
+
+                        <lookup-component
+                            :attribute="{'code': 'actions[' + index + '][value]', 'name': matchedAttribute.name, 'lookup_type': matchedAttribute.lookup_type}"
+                            validations="required"
+                            v-else
+                        ></lookup-component>
                     </div>
 
                     <div class="form-group" v-if="matchedAttribute.type == 'multiselect' || matchedAttribute.type == 'checkbox'">
diff --git a/packages/Webkul/Admin/src/Resources/views/settings/workflows/edit.blade.php b/packages/Webkul/Admin/src/Resources/views/settings/workflows/edit.blade.php
index 5dd701482..99ba99d97 100644
--- a/packages/Webkul/Admin/src/Resources/views/settings/workflows/edit.blade.php
+++ b/packages/Webkul/Admin/src/Resources/views/settings/workflows/edit.blade.php
@@ -328,11 +328,18 @@
                     </div>
 
                     <div class="form-group" v-if="matchedAttribute.type == 'select' || matchedAttribute.type == 'radio' || matchedAttribute.type == 'lookup'">
-                        <select :name="['actions[' + index + '][value]']" class="control" v-model="action.value">
+                        <select :name="['actions[' + index + '][value]']" class="control" v-model="action.value" v-if="! matchedAttribute.lookup_type">
                             <option v-for='option in matchedAttribute.options' :value="option.id">
                                 @{{ option.name }}
                             </option>
                         </select>
+
+                        <lookup-component
+                            :attribute="{'code': 'actions[' + index + '][value]', 'name': matchedAttribute.name, 'lookup_type': matchedAttribute.lookup_type}"
+                            validations="required"
+                            :data="action.value"
+                            v-else
+                        ></lookup-component>
                     </div>
 
                     <div class="form-group" v-if="matchedAttribute.type == 'multiselect' || matchedAttribute.type == 'checkbox'">
diff --git a/packages/Webkul/Workflow/src/Helpers/Entity/Lead.php b/packages/Webkul/Workflow/src/Helpers/Entity/Lead.php
index c461a801e..f2d6b0956 100644
--- a/packages/Webkul/Workflow/src/Helpers/Entity/Lead.php
+++ b/packages/Webkul/Workflow/src/Helpers/Entity/Lead.php
@@ -108,6 +108,29 @@ public function getEntity($entity)
         return $entity;
     }
 
+    /**
+     * Returns attributes
+     *
+     * @param  string  $entityType
+     * @param  array  $skipAttributes
+     * @return array
+     */
+    public function getAttributes($entityType, $skipAttributes = ['textarea', 'image', 'file', 'address'])
+    {
+        $attributes[] = [
+            'id'          => 'lead_pipeline_stage_id',
+            'type'        => 'select',
+            'name'        => 'Stage',
+            'lookup_type' => 'lead_pipeline_stages',
+            'options'     => collect([]),
+        ];
+
+        return array_merge(
+            parent::getAttributes($entityType, $skipAttributes),
+            $attributes
+        );
+    }
+
     /**
      * Returns workflow actions
      *