If it had a normal infinite switch, a two prong element would work. Since it's a pushbutton switch, I would go back to four prong, three wire element. If you could figure out how to wire a two prong to the pushbutton switch without shorting it out, you would only get two levels of heat, low and high.