Unit test doesn't test properly.