Fix beta cutoff bug