fix for waterfall plot signature change