Fix check element method

What does this MR do and why?

Fix the set_visibility method on jh because of an upstream change in the check_element method

MR acceptance checklist

This checklist encourages us to confirm any changes have been analyzed to reduce risks in quality, performance, reliability, security, and maintainability.

• I have evaluated the MR acceptance checklist for this MR.

/cc @prajnamas