Fixed broken integration