Added unit test - not working