Add missing flush