diff --git a/action.yml b/action.yml
index 4f006a1..fba1fba 100644
--- a/action.yml
+++ b/action.yml
@@ -5,12 +5,8 @@ inputs:
     description: 'rpc'
     required: true
     type: string
-  pid:
-    description: 'program PID'
-    required: true
-    type: string
-  pid_env_var:
-    description: 'program PID env var'
+  feature:
+    description: 'Feature to build'
     required: true
     type: string
   ssh_private_key:
@@ -75,12 +71,7 @@ runs:
     - run: chmod +x ~/.cargo/bin/anchor
       shell: bash
 
-    - name: Set program PID to devnet PID
-      run: |
-        echo "${{ inputs.pid_env_var }}=${{ inputs.pid }}" >> $GITHUB_ENV
-      shell: bash
-
     - name: Build
       run: |
-        anchor build
+        anchor build --features=${{ inputs.feature }}
       shell: bash