Polar heatmap with PyPlot works in REPL but not in IJulia

Plotting polar heatmap works in REPL but not in IJulia, here’s the MWE:

using PyPlot
r = [rd for rd = range(0.0, 10, length = 100), theta = range(0, 2π, length = 100)]
th = [theta for rd = range(0.0, 10, length = 100), theta = range(0, 2π, length = 100)]
subplot(projection="polar");
pcolormesh(th, r, r)

REPL produces the correct result:
repl

and IJulia a rectangular heatmap:
IJulia

Is there any tricks to solve this?

Agh. Putting all plotting commands to the same IJulia cell fixed the problem

Of course…