Figure 5 bug fix