fix train chapter 3