Bugfix to plots