Fix A/B testing issue