Update bug in Unit test