Remove $entity from flush method