Fix flag issue introduced by previous commit