Fix integration like tests