fixing figure 4