fix plotting