diff --git a/test/integration/foreman_puppet/host_js_test.rb b/test/integration/foreman_puppet/host_js_test.rb
index 875a5cab..7d530be7 100644
--- a/test/integration/foreman_puppet/host_js_test.rb
+++ b/test/integration/foreman_puppet/host_js_test.rb
@@ -23,7 +23,7 @@ class HostJSTest < IntegrationTestWithJavascript
         wait_for_ajax
         click_on_inherit('puppet_attributes_environment')
         select2(overridden_hostgroup.name, from: 'host_hostgroup_id')
-        assert page.select2_selector('host_puppet_attributes_environment_id').has_text? overridden_hostgroup.puppet.environment.name
+        assert page.find(select2_selector('host_puppet_attributes_environment_id')).has_text? overridden_hostgroup.puppet.environment.name
       end
 
       test 'sets fields to "inherit" when hostgroup is selected' do
@@ -145,7 +145,7 @@ class HostJSTest < IntegrationTestWithJavascript
         assert select2_chosen_selector('host_puppet_attributes_environment_id').has_text? original_hostgroup.puppet.environment.name
 
         # On host group change, the disabled select will be reset to an empty value - disabled select2 is invisible on chrome
-        assert select2_chosen_selector('host_puppet_proxy_id', visible: :all).has_text? ''
+        assert find(select2_selector('host_puppet_proxy_id'), visible: :all).has_text? ''
       end
 
       context 'has inherited Puppetclasses' do