Minor Fix in Unit Test