Fix for #376. The axis handles that the legend script uses are made old if no facets are used. This change updates the axs variable to point to the new axis handles