Fix integ tests