Fix AB testing code
Just an AB test code