[Ch7] Fix error in Unit Test