Fixes warning in Unit Test