Bug fix to prevent import error with missing matplotlib.pylab