FIX: Missing import in plots
Merge pull request #170 from danielballan/fix-missing-import

FIX: Missing import in plots