Fix issue with plotting.