Fixed matplotlib issue