diff --git a/src/plugins/ovirt-hosted-engine-setup/engine/add_host.py b/src/plugins/ovirt-hosted-engine-setup/engine/add_host.py
index b87d60ce..67c220cc 100644
--- a/src/plugins/ovirt-hosted-engine-setup/engine/add_host.py
+++ b/src/plugins/ovirt-hosted-engine-setup/engine/add_host.py
@@ -791,7 +791,7 @@ def _closeup(self):
                         id=mgmt_network_id
                     )
                     mgmt_network.set_vlan(
-                        self._ovirtsdk_xml.params.Vlan(id=vlan_id)
+                        self._ovirtsdk_xml.params.VLAN(id=vlan_id)
                     )
                     mgmt_network.update()
                     self._wait_network_vlan_ready(
@@ -823,7 +823,7 @@ def _closeup(self):
                             ohostedcons.NetworkEnv.HOST_NAME
                         ],
                         cluster=cluster,
-                        ssh=self._ovirtsdk_xml.params.Ssh(
+                        ssh=self._ovirtsdk_xml.params.SSH(
                             authentication_method='publickey',
                             port=self.environment[
                                 ohostedcons.NetworkEnv.SSHD_PORT