Fixed tests for chapter 5