Minor change to integration