Revert "Update flag.py"

This reverts commit 84115c1578e47820f98b2fe6071545ab1002023f.