Fix for FlagsController spec