minor changes to allow for new snap_integrator