Fixed up bad integration piece