Not working unit test