Fixes Unit test