matplotlib import fixes