unit test working fine now