fix import waffle flag