Move visibility check to page object