Fix up integration specs