Add test for when a flag is passed.