Fix error in UCB1 algorithm