Compare ids instead of objects to account for different batches