Fixes warnings in Unit Tests