BUG: Unrefactored parameter name change. I broke the plotting