Unit test finally working