diff --git a/roles/infra/roles/hosts/README.md b/roles/infra/roles/hosts/README.md
index d1d3008a..bf4842e1 100644
--- a/roles/infra/roles/hosts/README.md
+++ b/roles/infra/roles/hosts/README.md
@@ -19,6 +19,8 @@ The `hosts` list can contain the following parameters:
 | timeout       | 1200             | Maximum wait time for the host to be in an UP state.  |
 | poll_interval | 20               | Polling interval to check the host status. |
 | hosted_engine | UNDEF            | Specifies whether to 'deploy' or 'undeploy' hosted-engine to node. |
+| reboot_after_installation | UNDEF | If true reboot host after successful installation. |
+| reboot_after_upgrade | UNDEF | If true reboot host after successful upgrade. |
 
 Example Playbook
 ----------------
diff --git a/roles/infra/roles/hosts/tasks/main.yml b/roles/infra/roles/hosts/tasks/main.yml
index 6a39dc9e..16702e1e 100644
--- a/roles/infra/roles/hosts/tasks/main.yml
+++ b/roles/infra/roles/hosts/tasks/main.yml
@@ -37,6 +37,8 @@
     timeout: "{{ item.timeout | default(ovirt_hosts_add_timeout) }}"
     poll_interval: "{{ item.poll_interval | default(20) }}"
     hosted_engine: "{{ item.hosted_engine | default(omit) }}"
+    reboot_after_installation: "{{ item.reboot_after_installation | default(omit) }}"
+    reboot_after_upgrade: "{{ item.reboot_after_upgrade | default(omit) }}"
   with_items: "{{ ovirt_infra_hosts }}"
   loop_control:
     label: "{{ item.name }}"