wzb modification the logic