Changed component wrapper to component provider