diff --git a/playbooks/modify-package-state.yml b/playbooks/modify-package-state.yml
index 06d3de3..a162744 100644
--- a/playbooks/modify-package-state.yml
+++ b/playbooks/modify-package-state.yml
@@ -1,5 +1,5 @@
 ---
-- hosts: all
+- hosts: "{{ tag }}"
   become: true
   vars_files:
     - ../secret
@@ -11,7 +11,7 @@
       prompt: "State"
       private: no
     - name: tag
-      hosts: "hosts"
+      prompt: "hosts"
       private: no
   tasks:
     - name: Modify state of package in debian and ubuntu
@@ -21,5 +21,3 @@
         state: "{{ state }}"
         autoremove: yes
       when: ansible_facts['os_family'] == "Debian"
-      hosts: 
-        - "{{ tag }}"