Bug fix model plot