Fix for plot limits