Add class checks to avoid conflict with cocoon