minor bug fix after completing Chapter 7