Fixed bug in AB test