Fixed plugin for new plotting API