Fix Figure 4