Add beta to apply_pde_operator