small changes for the integration