Double check all your >= and <=.
I did this in 4 separate lines to keep it simple, but most people seem to do it in one line as you have.
The savefig solution is interesting, but I think the tests won’t work with that. You can achieve the same by adding the line
fig = fig.figure
just before the savefig. You can read a little more about this here:
I hope this helps!