Fixes pylab import issue on Ubuntu.