Add a test for the feature flags