Fix update_visibility