Added missing flush method