Fixed a little bit on the test chapter