Code cleanup
Add missing flag and remove duplicate

Thanks to @andypsylon for pointing it out