Remove matplotlib mock?