bug fix for plotting associations