fix beta_only handling