fix chapter 2 with lightweight