add the flag i guess
small refactor