minor correction to pbe solution