ABTest bug fix