Fixes a misplaced import

misplaced MatplolibWidget import