fix feature flag check