More conserative with importing matplotlib